Update app.py

app.py

@@ -1,32 +1,39 @@
 import gradio as gr
 from transformers import pipeline
 
-# Load a free language model pipeline
-# Replace 'google/flan-t5-small' with another open-source model if needed
-strategy_generator = pipeline("text2text-generation", model="google/flan-t5-small")
+# Load a more capable open-source model
+# You can use "google/flan-t5-large" or other alternatives depending on available resources
+strategy_generator = pipeline("text2text-generation", model="google/flan-t5-large")
 
 # Function to generate a business strategy
 def generate_strategy(industry, challenge, goals):
     prompt = f"""
-    Create a detailed business strategy for the following:
-    Industry: {industry}
-    Challenge: {challenge}
-    Goals: {goals}
+    Generate a detailed and actionable business strategy for a company in the {industry} industry.
+    - Challenge: {challenge}
+    - Goals: {goals}
+    Provide realistic steps and measurable outcomes to address the challenge and achieve the goals.
     """
-    response = strategy_generator(prompt, max_length=200, num_return_sequences=1)
-    return response[0]['generated_text']
+    try:
+        response = strategy_generator(prompt, max_length=300, num_return_sequences=1)
+        return response[0]['generated_text']
+    except Exception as e:
+        return f"Error generating strategy: {e}"
 
 # Function to perform a SWOT analysis
 def swot_analysis(strengths, weaknesses, opportunities, threats):
     prompt = f"""
-    Perform a SWOT analysis based on the following:
-    Strengths: {strengths}
-    Weaknesses: {weaknesses}
-    Opportunities: {opportunities}
-    Threats: {threats}
+    Perform a comprehensive SWOT analysis for a company based on the following:
+    - Strengths: {strengths}
+    - Weaknesses: {weaknesses}
+    - Opportunities: {opportunities}
+    - Threats: {threats}
+    Provide actionable insights for each category to guide business decision-making.
     """
-    response = strategy_generator(prompt, max_length=200, num_return_sequences=1)
-    return response[0]['generated_text']
+    try:
+        response = strategy_generator(prompt, max_length=300, num_return_sequences=1)
+        return response[0]['generated_text']
+    except Exception as e:
+        return f"Error generating SWOT analysis: {e}"
 
 # Gradio interface
 with gr.Blocks() as demo: