Update app.py

app.py

@@ -12,12 +12,29 @@ from transformers import pipeline
 logging.basicConfig(level=logging.DEBUG, format='%(asctime)s - %(levelname)s - %(message)s')
 
 # =============================================================================
-# Load In-House Hugging Face Models for AI Agents
+# Load In-House Hugging Face Models for AI Agents with Fallbacks
 # =============================================================================
-# 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)
+# Replace these model IDs with your actual in-house model names and token if available.
+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)
 
 # =============================================================================
 # Helper Functions: Interactive Chart Generators
@@ -378,42 +395,10 @@ def display_pitch_deck_overview():
     """
     return pitch_deck
 
-# =============================================================================
-# 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
 # -----------------------------------------------------------------------------
@@ -610,7 +595,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css="""
                 """)
             
             # -------------------------------------------------------------------------
-            # Tab 8: SaaS Solutions & Revenue Model
+            # Tab 8: SaaS Solutions, Systems & Revenue Model
             # -------------------------------------------------------------------------
             with gr.TabItem("🖥️ SaaS Solutions & Revenue Model"):
                 gr.Markdown("""