Update app.py

app.py

@@ -4,7 +4,7 @@ from transformers import pipeline
 # Load a more capable open-source model
 strategy_generator = pipeline("text2text-generation", model="google/flan-t5-large")
 
-# Function to clean up repetitive sentences in generated text
+# Function to clean up repetitive or irrelevant sentences in generated text
 def clean_output(text):
     sentences = text.split(". ")
     seen = set()
@@ -18,13 +18,20 @@ def clean_output(text):
 # Function to generate a business strategy
 def generate_strategy(industry, challenge, goals):
     prompt = f"""
-    You are a business consultant helping a company in the {industry} industry.
+    You are a professional business consultant specializing in the {industry} industry.
     The company faces the following challenge: {challenge}.
     The company aims to achieve the following goal: {goals}.
-    Provide a detailed and actionable business strategy with specific steps, measurable outcomes, and innovative ideas.
+    Provide a detailed business strategy including:
+    - 3-5 actionable steps
+    - Measurable short-term outcomes (6 months to 1 year)
+    - Innovative ideas tailored to the industry
+    Example format:
+    1. Step 1: Description
+    2. Step 2: Description
+    3. Step 3: Description
     """
     try:
-        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7)
+        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7, top_p=0.9)
         cleaned_response = clean_output(response[0]['generated_text'])
         return cleaned_response
     except Exception as e:
@@ -33,15 +40,15 @@ def generate_strategy(industry, challenge, goals):
 # Function to perform a SWOT analysis
 def swot_analysis(strengths, weaknesses, opportunities, threats):
     prompt = f"""
-    You are a business consultant performing a SWOT analysis for a company.
+    You are a professional business consultant performing a SWOT analysis for a company.
     - Strengths: {strengths}
     - Weaknesses: {weaknesses}
     - Opportunities: {opportunities}
     - Threats: {threats}
-    Provide actionable insights for each category and suggestions on how to maximize strengths and opportunities while addressing weaknesses and threats.
+    Provide actionable insights for each category, including specific recommendations on how to leverage strengths and opportunities while addressing weaknesses and threats.
     """
     try:
-        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7)
+        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7, top_p=0.9)
         cleaned_response = clean_output(response[0]['generated_text'])
         return cleaned_response
     except Exception as e: