Update app.py

app.py

@@ -4,16 +4,108 @@ import requests
 from datetime import datetime, timedelta
 import logging
 import random
+from transformers import pipeline
+from huggingface_hub import InferenceClient
+from PIL import Image
+import io
+import os
 
 # =============================================================================
-# Setup Logging
+# Setup Logging and API Keys
 # =============================================================================
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
+# Image Generation Setup
+API_KEY = os.getenv("HF_API_TOKEN")
+MODEL_NAME = "black-forest-labs/FLUX.1-dev"
+client = InferenceClient(api_key=API_KEY)
+
 # =============================================================================
-# Helper Functions: Interactive Chart Generators
+# Load In-House Hugging Face Models for AI Agents with Fallbacks
+# =============================================================================
+try:
+    summarizer = pipeline("summarization", model="myorg/inhouse-summarizer")
+    logging.info("Loaded in-house summarizer.")
+except Exception as e:
+    logging.error("In-house summarizer not found, falling back: %s", e)
+    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
+
+try:
+    research_generator = pipeline("text-generation", model="myorg/inhouse-research", max_length=100)
+    logging.info("Loaded in-house research generator.")
+except Exception as e:
+    logging.error("In-house research generator not found, falling back: %s", e)
+    research_generator = pipeline("text-generation", model="gpt2", max_length=100)
+
+try:
+    planner_generator = pipeline("text-generation", model="myorg/inhouse-planner", max_length=150)
+    logging.info("Loaded in-house planner generator.")
+except Exception as e:
+    logging.error("In-house planner generator not found, falling back: %s", e)
+    planner_generator = pipeline("text-generation", model="gpt2", max_length=150)
+
+# =============================================================================
+# In-House Agent Functions
+# =============================================================================
+def report_agent(data):
+    """
+    Generates a summary report using the in-house summarizer model.
+    """
+    if not data or len(data.strip()) == 0:
+        return "Please provide input text for summarization."
+    try:
+        summary = summarizer(data, max_length=130, min_length=30, do_sample=False)
+        logging.debug("Summary generated using in-house summarizer.")
+        return summary[0]['summary_text']
+    except Exception as e:
+        logging.error("Error generating summary: %s", e)
+        return "Error generating summary."
+
+def planning_agent(goal):
+    """
+    Generates a detailed action plan based on the provided goal using the in-house planner model.
+    """
+    if not goal or len(goal.strip()) == 0:
+        return "Please provide a goal for planning."
+    try:
+        plan = planner_generator(goal, max_length=150, num_return_sequences=1)
+        logging.debug("Plan generated using in-house planner.")
+        return plan[0]['generated_text']
+    except Exception as e:
+        logging.error("Error generating plan: %s", e)
+        return "Error generating plan."
+
+def research_agent(query):
+    """
+    Generates research insights based on the provided query using the in-house research generator.
+    """
+    if not query or len(query.strip()) == 0:
+        return "Please provide a research query."
+    try:
+        result = research_generator(query, max_length=100, num_return_sequences=1)
+        logging.debug("Research output generated using in-house research model.")
+        return result[0]['generated_text']
+    except Exception as e:
+        logging.error("Error generating research output: %s", e)
+        return "Error generating research output."
+
 # =============================================================================
+# Image Generation Function
+# =============================================================================
+def generate_image(prompt):
+    """Generate an image from a text prompt using Hugging Face API."""
+    try:
+        image = client.text_to_image(prompt, model=MODEL_NAME)
+        img_byte_array = io.BytesIO()
+        image.save(img_byte_array, format="PNG")
+        img_byte_array.seek(0)
+        return image
+    except Exception as e:
+        return f"Error generating image: {e}"
 
+# =============================================================================
+# Helper Functions: Interactive Chart Generators
+# =============================================================================
 def create_performance_comparison():
     """
     Creates a performance comparison chart showing current (2GB VRAM) capabilities
@@ -107,7 +199,6 @@ def create_earnings_history():
 def calculate_loan_schedule(loan_amount=3000, interest_rate=5.0):
     """
     Calculates the loan repayment schedule given an investment amount and interest rate.
-    Returns a formatted string detailing monthly payments and timelines.
     """
     try:
         amount = float(loan_amount)
@@ -145,42 +236,6 @@ def calculate_loan_schedule(loan_amount=3000, interest_rate=5.0):
     except ValueError:
         return "Please enter valid numbers for the loan calculation."
 
-def create_detailed_financial_projection():
-    """
-    Creates a detailed 12-month financial projection chart with three scenarios:
-    worst-case, expected, and best-case revenue paths.
-    """
-    months = ['Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec', 'Jan', 'Feb']
-    base_revenue = 1000
-    worst_case = [base_revenue * (1 + 0.05) ** i for i in range(len(months))]
-    expected = [base_revenue * (1 + 0.10) ** i for i in range(len(months))]
-    best_case = [base_revenue * (1 + 0.20) ** i for i in range(len(months))]
-    
-    fig = go.Figure()
-    fig.add_trace(go.Scatter(
-        x=months, y=worst_case, name='Worst Case',
-        line={'color': '#FF6B6B', 'width': 2, 'dash': 'dot'}
-    ))
-    fig.add_trace(go.Scatter(
-        x=months, y=expected, name='Expected',
-        line={'color': '#94A3B8', 'width': 3}
-    ))
-    fig.add_trace(go.Scatter(
-        x=months, y=best_case, name='Best Case',
-        line={'color': '#2563EB', 'width': 3, 'dash': 'dash'}
-    ))
-    
-    fig.update_layout(
-        title={'text': '12-Month Financial Projection', 'font': {'size': 24}},
-        xaxis_title='Month',
-        yaxis_title='Revenue (USD)',
-        plot_bgcolor='white',
-        font={'family': 'Arial', 'size': 14},
-        legend={'orientation': 'h', 'x': 0.3, 'y': 1.1}
-    )
-    logging.debug("Detailed financial projection chart created.")
-    return fig
-
 def create_project_timeline():
     """
     Creates a timeline chart showing key project milestones.
@@ -189,8 +244,12 @@ def create_project_timeline():
     dates = ["2025-03-01", "2025-04-15", "2025-06-01", "2025-08-01", "2025-10-01", "2025-12-01"]
     
     fig = go.Figure(data=go.Scatter(
-        x=dates, y=list(range(len(milestones))), mode="markers+text",
-        text=milestones, textposition="top center", marker=dict(size=12, color="#2563EB")
+        x=dates,
+        y=list(range(len(milestones))),
+        mode="markers+text",
+        text=milestones,
+        textposition="top center",
+        marker=dict(size=12, color="#2563EB")
     ))
     fig.update_layout(
         title="Project Timeline & Milestones",
@@ -206,7 +265,6 @@ def create_project_timeline():
 # =============================================================================
 # Helper Functions: Textual Content for Pitch Deck
 # =============================================================================
-
 def display_tech_stack():
     """
     Returns a Markdown string detailing the technical stack and tools used.
@@ -289,7 +347,7 @@ def display_personal_note():
     - Create both proprietary and open source RAG systems.
     - Operate a marketing advertising firm with automated agents.
     
-    This GPU upgrade is not just a hardware upgrade—it’s the key to turning my passion into a sustainable business
+    This GPU upgrade is not just a hardware upgrade—it's the key to turning my passion into a sustainable business
     that repays your investment and makes you proud.
     """
     return personal_note
@@ -511,7 +569,37 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
                 gr.Markdown(display_pitch_deck_overview())
             
             # -------------------------------------------------------------------------
-            # Tab 5: Project Timeline & Milestones
+            # Tab 5: 🤖 AI Agents
+            # -------------------------------------------------------------------------
+            with gr.TabItem("🤖 AI Agents"):
+                gr.Markdown("""
+                    ### AI Agent Tools
+                    Use these tools to generate summaries, action plans, and research insights
+                    using our in-house Hugging Face models.
+                """)
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("#### Summarizer")
+                        summarizer_input = gr.Textbox(label="Enter text to summarize", placeholder="Paste text here...", lines=5)
+                        summarizer_output = gr.Textbox(label="Summary", lines=5)
+                        summarizer_button = gr.Button("Generate Summary")
+                        summarizer_button.click(report_agent, inputs=[summarizer_input], outputs=[summarizer_output])
+                    with gr.Column():
+                        gr.Markdown("#### Planner")
+                        planner_input = gr.Textbox(label="Enter goal or task", placeholder="Describe your goal...", lines=5)
+                        planner_output = gr.Textbox(label="Action Plan", lines=5)
+                        planner_button = gr.Button("Generate Plan")
+                        planner_button.click(planning_agent, inputs=[planner_input], outputs=[planner_output])
+                with gr.Row():
+                    with gr.Column():
+                        gr.Markdown("#### Research Agent")
+                        research_input = gr.Textbox(label="Enter research query", placeholder="Ask your research question...", lines=5)
+                        research_output = gr.Textbox(label="Research Insights", lines=5)
+                        research_button = gr.Button("Generate Research")
+                        research_button.click(research_agent, inputs=[research_input], outputs=[research_output])
+            
+            # -------------------------------------------------------------------------
+            # Tab 6: Project Timeline & Milestones
             # -------------------------------------------------------------------------
             with gr.TabItem("📅 Project Timeline & Milestones"):
                 gr.Markdown("""
@@ -529,7 +617,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
                 """)
             
             # -------------------------------------------------------------------------
-            # Tab 6: From the Heart & Contact
+            # Tab 7: From the Heart & Contact
             # -------------------------------------------------------------------------
             with gr.TabItem("💝 From the Heart"):
                 gr.Markdown(display_personal_note())
@@ -539,9 +627,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
                     - **LinkedIn:** <a href="https://www.linkedin.com/in/waynesletcher/" target="_blank" style="color: #2563EB;">Wayne Sletcher</a>
                     - **Website:** <a href="https://www.sletchersystems.com/" target="_blank" style="color: #2563EB;">sletchersystems.com</a>
                 """)
-        
+            
             # -------------------------------------------------------------------------
-            # Tab 7: SaaS Solutions, Systems & Revenue Model
+            # Tab 8: SaaS Solutions, Systems & Revenue Model
             # -------------------------------------------------------------------------
             with gr.TabItem("🖥️ SaaS Solutions & Revenue Model"):
                 gr.Markdown("""
@@ -583,8 +671,9 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
 #       - Company Purpose, Problem, Solution, Why Now, Market Size, Competition,
 #         Product, Business Model, Team, and Financials
 #   - An in-depth look at my technical stack and current projects
+#   - A new tab ("🤖 AI Agents") for in-house NLP tasks: summarization, planning, research
 #   - A project timeline and milestones for future growth
-#   - A heartfelt personal note aimed at making Dad proud
+#   - A heartfelt personal note aimed at making Dad proud, along with contact details
 #   - Persistent call-to-action with contact details (email, LinkedIn, website)
 #
 # Future enhancements may include:
@@ -597,152 +686,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
 #
 # =============================================================================
 #
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
-#
 # =============================================================================
 # End of File
 # =============================================================================