Update app.py

app.py

@@ -10,47 +10,52 @@ def clean_output(text):
     seen = set()
     cleaned_sentences = []
     for sentence in sentences:
-        if sentence not in seen:
-            seen.add(sentence)
-            cleaned_sentences.append(sentence)
+        if sentence.strip() not in seen:
+            seen.add(sentence.strip())
+            cleaned_sentences.append(sentence.strip())
     return ". ".join(cleaned_sentences)
 
+# Function to validate and refine generated text
+def validate_output(output, prompt):
+    if len(output.split()) < 10 or output.lower().startswith("the company") or output.count("The company") > 2:
+        return "The response was unclear. Please adjust the input or try again with a refined question."
+    return clean_output(output)
+
 # Function to generate a business strategy
 def generate_strategy(industry, challenge, goals):
     prompt = f"""
-    You are a professional business consultant specializing in the {industry} industry.
+    You are an experienced 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 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
+    The company's goal is to achieve: {goals}.
+    Provide a strategy with:
+    1. Three to five actionable steps.
+    2. Measurable outcomes within the next 6 months to 1 year.
+    Example:
+    1. Improve marketing through targeted ads.
+    2. Streamline operations for better efficiency.
+    3. Partner with local businesses to reach new customers.
     """
     try:
         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
+        validated_response = validate_output(response[0]['generated_text'], prompt)
+        return validated_response
     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"""
-    You are a professional business consultant performing a SWOT analysis for a company.
+    You are an experienced business consultant performing a SWOT analysis for a company.
     - Strengths: {strengths}
     - Weaknesses: {weaknesses}
     - Opportunities: {opportunities}
     - Threats: {threats}
-    Provide actionable insights for each category, including specific recommendations on how to leverage strengths and opportunities while addressing weaknesses and threats.
+    Provide detailed insights for each category and specific recommendations for improvement.
     """
     try:
         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
+        validated_response = validate_output(response[0]['generated_text'], prompt)
+        return validated_response
     except Exception as e:
         return f"Error generating SWOT analysis: {e}"