Update app.py

app.py

@@ -15,47 +15,41 @@ def clean_output(text):
             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 an experienced business consultant specializing in the {industry} industry.
+    You are a business consultant with expertise in the {industry} industry.
     The company faces the following challenge: {challenge}.
     The company's goal is to achieve: {goals}.
-    Provide a strategy with:
+    Provide a detailed strategy that includes:
     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.
+    2. For each step, explain WHY it is recommended and HOW to implement it.
+    Example format:
+    1. Actionable Step: Description
+       - Why: Reasoning for this step.
+       - How: Implementation details.
     """
     try:
-        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7, top_p=0.9)
-        validated_response = validate_output(response[0]['generated_text'], prompt)
-        return validated_response
+        response = strategy_generator(prompt, max_length=500, 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:
         return f"Error generating strategy: {e}"
 
 # Function to perform a SWOT analysis
 def swot_analysis(strengths, weaknesses, opportunities, threats):
     prompt = f"""
-    You are an experienced business consultant performing a SWOT analysis for a company.
+    You are a business consultant performing a SWOT analysis for a company.
     - Strengths: {strengths}
     - Weaknesses: {weaknesses}
     - Opportunities: {opportunities}
     - Threats: {threats}
-    Provide detailed insights for each category and specific recommendations for improvement.
+    Provide detailed insights for each category, and include specific recommendations for improvement.
     """
     try:
         response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7, top_p=0.9)
-        validated_response = validate_output(response[0]['generated_text'], prompt)
-        return validated_response
+        cleaned_response = clean_output(response[0]['generated_text'])
+        return cleaned_response
     except Exception as e:
         return f"Error generating SWOT analysis: {e}"