Update app.py

app.py

@@ -2,10 +2,10 @@ import gradio as gr
 from transformers import pipeline
 
 # Load a free language model pipeline
-# You can replace 'google/flan-t5-small' with other free models like 'bigscience/bloom-560m'
+# Replace 'google/flan-t5-small' with another open-source model if needed
 strategy_generator = pipeline("text2text-generation", model="google/flan-t5-small")
 
-# Function to generate business strategy
+# Function to generate a business strategy
 def generate_strategy(industry, challenge, goals):
     prompt = f"""
     Create a detailed business strategy for the following:
@@ -16,7 +16,7 @@ def generate_strategy(industry, challenge, goals):
     response = strategy_generator(prompt, max_length=200, num_return_sequences=1)
     return response[0]['generated_text']
 
-# Function to perform SWOT analysis
+# Function to perform a SWOT analysis
 def swot_analysis(strengths, weaknesses, opportunities, threats):
     prompt = f"""
     Perform a SWOT analysis based on the following:
@@ -31,9 +31,11 @@ def swot_analysis(strengths, weaknesses, opportunities, threats):
 # Gradio interface
 with gr.Blocks() as demo:
     gr.Markdown("# AI Business Strategy Generator")
-    gr.Markdown("Create actionable business strategies and SWOT analyses.")
+    gr.Markdown("Generate actionable business strategies and SWOT analyses using AI.")
 
+    # Tab 1: Generate Business Strategy
     with gr.Tab("Generate Strategy"):
+        gr.Markdown("### Input Information to Generate a Business Strategy")
         industry_input = gr.Textbox(label="Industry", placeholder="E.g., E-commerce, Healthcare")
         challenge_input = gr.Textbox(label="Key Challenge", placeholder="E.g., Low customer retention")
         goals_input = gr.Textbox(label="Goals", placeholder="E.g., Increase sales by 20% in 6 months")
@@ -46,6 +48,21 @@ 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="Weak
+        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()