Update app.py

app.py

@@ -4,6 +4,7 @@ import requests
 from datetime import datetime, timedelta
 import logging
 import random
+from transformers import pipeline
 
 # =============================================================================
 # Setup Logging
@@ -11,9 +12,16 @@ import random
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # =============================================================================
-# Helper Functions: Interactive Chart Generators
+# Load In-House Hugging Face Models for AI Agents
 # =============================================================================
+# Replace these model IDs with your actual in-house model names.
+summarizer = pipeline("summarization", model="myorg/inhouse-summarizer")
+research_generator = pipeline("text-generation", model="myorg/inhouse-research", max_length=100)
+planner_generator = pipeline("text-generation", model="myorg/inhouse-planner", max_length=150)
 
+# =============================================================================
+# Helper Functions: Interactive Chart Generators
+# =============================================================================
 def create_performance_comparison():
     """
     Creates a performance comparison chart showing current (2GB VRAM) capabilities
@@ -206,7 +214,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.
@@ -372,9 +379,41 @@ def display_pitch_deck_overview():
     return pitch_deck
 
 # =============================================================================
-# End of Helper Functions
+# New Agent Functions Using In-House Hugging Face Models
 # =============================================================================
+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."
+    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']
 
+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."
+    plan = planner_generator(goal, max_length=150, num_return_sequences=1)
+    logging.debug("Plan generated using in-house planner.")
+    return plan[0]['generated_text']
+
+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."
+    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']
+
+# =============================================================================
+# End of Helper Functions
+# =============================================================================
 # -----------------------------------------------------------------------------
 # Construct the Gradio Interface with Multiple Tabs
 # -----------------------------------------------------------------------------
@@ -511,7 +550,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 +598,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 +608,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 & Revenue Model
             # -------------------------------------------------------------------------
             with gr.TabItem("🖥️ SaaS Solutions & Revenue Model"):
                 gr.Markdown("""
@@ -583,8 +652,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:
@@ -739,10 +809,6 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
 #
 #
 #
-#
-#
-#
-#
 # =============================================================================
 # End of File
 # =============================================================================