Update app.py

app.py

@@ -2,36 +2,48 @@ import gradio as gr
 from transformers import pipeline
 
 # 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 clean up repetitive sentences in generated text
+def clean_output(text):
+    sentences = text.split(". ")
+    seen = set()
+    cleaned_sentences = []
+    for sentence in sentences:
+        if sentence not in seen:
+            seen.add(sentence)
+            cleaned_sentences.append(sentence)
+    return ". ".join(cleaned_sentences)
+
 # Function to generate a business strategy
 def generate_strategy(industry, challenge, goals):
     prompt = f"""
-    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.
+    You are a business consultant helping a company 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.
     """
     try:
-        response = strategy_generator(prompt, max_length=300, num_return_sequences=1)
-        return response[0]['generated_text']
+        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7)
+        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"""
-    Perform a comprehensive SWOT analysis for a company based on the following:
+    You are a business consultant performing a SWOT analysis for a company.
     - Strengths: {strengths}
     - Weaknesses: {weaknesses}
     - Opportunities: {opportunities}
     - Threats: {threats}
-    Provide actionable insights for each category to guide business decision-making.
+    Provide actionable insights for each category and suggestions on how to maximize strengths and opportunities while addressing weaknesses and threats.
     """
     try:
-        response = strategy_generator(prompt, max_length=300, num_return_sequences=1)
-        return response[0]['generated_text']
+        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7)
+        cleaned_response = clean_output(response[0]['generated_text'])
+        return cleaned_response
     except Exception as e:
         return f"Error generating SWOT analysis: {e}"