Update app.py

app.py

@@ -8,7 +8,6 @@ openai.api_base = "https://api.groq.com/openai/v1"
 
 # Function to get response from GROQ API based on category keywords
 def get_groq_response(message, category):
-    # System message to tailor responses based on the category
     system_message = ""
     if category == "Stress Management":
         system_message = "Provide soothing advice and tips to help the user manage stress. Be calm and empathetic."
@@ -38,7 +37,6 @@ def recognize_category(user_input):
     general_keywords = ["hello", "hi", "how", "what", "why", "chat", "talk"]
     friendly_keywords = ["friend", "buddy", "hangout", "fun", "relax"]
 
-    # Check for category based on keywords
     if any(keyword in user_input.lower() for keyword in stress_keywords):
         return "Stress Management"
     elif any(keyword in user_input.lower() for keyword in career_keywords):
@@ -48,62 +46,45 @@ def recognize_category(user_input):
     elif any(keyword in user_input.lower() for keyword in friendly_keywords):
         return "Friendly Buddy"
     else:
-        return "General"  # Default category if no keywords match
+        return "General"  # Default category
 
-# Chatbot function
-def chatbot(user_input, history=[]):
-    category = recognize_category(user_input)  # Determine the category based on user input
+# Function to process input and generate category-specific responses
+def process_input(user_input, history):
+    category = recognize_category(user_input)
     bot_response = get_groq_response(user_input, category)
-    history.append((category, user_input, bot_response))  # Save category, user input, and bot response
-    return history
+    history.append((category, user_input, bot_response))
 
-# Gradio Interface setup with Blocks for custom layout
+    # Group responses by category
+    stress_responses = [f"**User**: {u}\n**Bot**: {b}" for c, u, b in history if c == "Stress Management"]
+    career_responses = [f"**User**: {u}\n**Bot**: {b}" for c, u, b in history if c == "Career Advice"]
+    general_responses = [f"**User**: {u}\n**Bot**: {b}" for c, u, b in history if c == "General"]
+    friendly_responses = [f"**User**: {u}\n**Bot**: {b}" for c, u, b in history if c == "Friendly Buddy"]
+
+    # Return grouped responses for display
+    return (
+        "\n\n".join(stress_responses) or "No responses yet.",
+        "\n\n".join(career_responses) or "No responses yet.",
+        "\n\n".join(general_responses) or "No responses yet.",
+        "\n\n".join(friendly_responses) or "No responses yet.",
+        history
+    )
+
+# Gradio UI with Blocks
 with gr.Blocks() as demo:
-    gr.Markdown("## Meet your Personal Assistant: Your Helpful and Caring Chatbot")
-    gr.Markdown("This chatbot is here to help you with anything you need, whether it’s answering questions, offering support, or guiding you through tasks. With a friendly and empathetic approach, it listens carefully to your concerns and provides thoughtful, understanding responses.")
+    gr.Markdown("## Meet Your Personal Assistant: A Caring Chatbot")
+    gr.Markdown("This chatbot helps with stress management, career advice, general queries, or just being a friendly buddy!")
 
     with gr.Row():
         user_input = gr.Textbox(label="Your Message", placeholder="Type your message here...")
         submit_button = gr.Button("Submit")
-    
-    history = gr.State([])
-    stress_column = gr.Column()
-    career_column = gr.Column()
-    general_column = gr.Column()
-    friendly_column = gr.Column()
-
-    output_area = gr.Row([stress_column, career_column, general_column, friendly_column])
 
-    def process_input(user_input, history):
-        category = recognize_category(user_input)
-        bot_response = get_groq_response(user_input, category)
-        history.append((category, user_input, bot_response))
-
-        # Organize responses by category
-        stress_responses = []
-        career_responses = []
-        general_responses = []
-        friendly_responses = []
-
-        for cat, user_msg, bot_msg in history:
-            response = f"**User**: {user_msg}\n**Bot**: {bot_msg}"
-            if cat == "Stress Management":
-                stress_responses.append(response)
-            elif cat == "Career Advice":
-                career_responses.append(response)
-            elif cat == "General":
-                general_responses.append(response)
-            elif cat == "Friendly Buddy":
-                friendly_responses.append(response)
+    with gr.Row():
+        stress_column = gr.Textbox(label="Stress Management", interactive=False)
+        career_column = gr.Textbox(label="Career Advice", interactive=False)
+        general_column = gr.Textbox(label="General", interactive=False)
+        friendly_column = gr.Textbox(label="Friendly Buddy", interactive=False)
 
-        # Update the columns
-        return {
-            stress_column: "\n\n".join(stress_responses) or "No responses yet.",
-            career_column: "\n\n".join(career_responses) or "No responses yet.",
-            general_column: "\n\n".join(general_responses) or "No responses yet.",
-            friendly_column: "\n\n".join(friendly_responses) or "No responses yet.",
-            history: history
-        }
+    history = gr.State([])
 
     submit_button.click(
         fn=process_input,