Update app.py

app.py

@@ -15,43 +15,53 @@ def clean_output(text):
             cleaned_sentences.append(sentence.strip())
     return ". ".join(cleaned_sentences)
 
-# Function to generate a business strategy
-def generate_strategy(industry, challenge, goals):
+# Function to generate actionable steps
+def generate_steps(industry, challenge, goals):
     prompt = f"""
     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 detailed strategy that includes:
-    1. Three to five actionable steps.
-    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.
+    Provide three to five actionable steps to help the company achieve this goal.
+    Focus on specific, realistic, and innovative strategies relevant to the industry.
     """
     try:
-        response = strategy_generator(prompt, max_length=500, num_return_sequences=1, temperature=0.7, top_p=0.9)
+        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:
-        return f"Error generating strategy: {e}"
+        return f"Error generating steps: {e}"
 
-# Function to perform a SWOT analysis
-def swot_analysis(strengths, weaknesses, opportunities, threats):
+# Function to generate "why" and "how" for each step
+def expand_step(step):
     prompt = f"""
-    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 include specific recommendations for improvement.
+    You are a business consultant. For the following strategy:
+    "{step}"
+    Provide:
+    - Why this step is recommended.
+    - How to implement this step effectively.
     """
     try:
-        response = strategy_generator(prompt, max_length=300, num_return_sequences=1, temperature=0.7, top_p=0.9)
+        response = strategy_generator(prompt, max_length=200, 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 SWOT analysis: {e}"
+        return f"Error expanding step: {e}"
+
+# Combined function to generate strategy
+def generate_strategy(industry, challenge, goals):
+    # Generate initial steps
+    steps = generate_steps(industry, challenge, goals)
+    if "Error" in steps:
+        return steps
+    
+    # Expand each step
+    steps_list = steps.split("\n")
+    detailed_steps = []
+    for step in steps_list:
+        if step.strip():
+            detailed_steps.append(f"{step}\n{expand_step(step)}")
+    
+    return "\n\n".join(detailed_steps)
 
 # Gradio interface
 with gr.Blocks() as demo:
@@ -73,21 +83,5 @@ with gr.Blocks() as demo:
             outputs=[strategy_output]
         )
 
-    # Tab 2: SWOT Analysis
-    with gr.Tab("SWOT Analysis"):
-        gr.Markdown("### Input Information to Perform a SWOT Analysis")
-        strengths_input = gr.Textbox(label="Strengths", placeholder="E.g., Strong brand presence")
-        weaknesses_input = gr.Textbox(label="Weaknesses", placeholder="E.g., Limited market reach")
-        opportunities_input = gr.Textbox(label="Opportunities", placeholder="E.g., Emerging markets")
-        threats_input = gr.Textbox(label="Threats", placeholder="E.g., Rising competition")
-        swot_button = gr.Button("Perform SWOT Analysis")
-        swot_output = gr.Textbox(label="Generated SWOT Analysis", lines=10)
-
-        swot_button.click(
-            swot_analysis,
-            inputs=[strengths_input, weaknesses_input, opportunities_input, threats_input],
-            outputs=[swot_output]
-        )
-
 # Launch the Gradio app
 demo.launch()