Update app.py

app.py

@@ -55,44 +55,60 @@ def chatbot(user_input, history=[]):
     category = recognize_category(user_input)  # Determine the category based on 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
+    return history
 
-# Gradio Interface setup with a vertical layout for category responses
-chat_interface = gr.Interface(
-    fn=chatbot,
-    inputs=[
-        gr.Textbox(label="Your Message"),  # User input
-        "state"  # History
-    ],
-    outputs=["chatbot", "state"],
-    live=False,
-    title="Meet your Personal Assistant: Your Helpful and Caring Chatbot",
-    description="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. Whether you’re seeking information or just someone to chat with, our chatbot is designed to make you feel heard and supported. It’s more than just a tool—it’s a companion dedicated to making your day easier and more enjoyable.",
-    layout="vertical"
-)
+# Gradio Interface setup with Blocks for custom layout
+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.")
 
-# Custom layout to show the outputs under different vertical columns based on categories
-def custom_output(history):
-    stress_responses = []
-    career_responses = []
-    general_responses = []
-    friendly_responses = []
+    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()
 
-    for category, user_input, bot_response in history:
-        if category == "Stress Management":
-            stress_responses.append(f"**User**: {user_input}\n**Bot**: {bot_response}")
-        elif category == "Career Advice":
-            career_responses.append(f"**User**: {user_input}\n**Bot**: {bot_response}")
-        elif category == "General":
-            general_responses.append(f"**User**: {user_input}\n**Bot**: {bot_response}")
-        elif category == "Friendly Buddy":
-            friendly_responses.append(f"**User**: {user_input}\n**Bot**: {bot_response}")
+    output_area = gr.Row([stress_column, career_column, general_column, friendly_column])
 
-    return [
-        gr.Column([gr.Markdown(f"### Stress Management\n{response}") for response in stress_responses]),
-        gr.Column([gr.Markdown(f"### Career Advice\n{response}") for response in career_responses]),
-        gr.Column([gr.Markdown(f"### General\n{response}") for response in general_responses]),
-        gr.Column([gr.Markdown(f"### Friendly Buddy\n{response}") for response in friendly_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))
 
-chat_interface.launch()
+        # 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)
+
+        # 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
+        }
+
+    submit_button.click(
+        fn=process_input,
+        inputs=[user_input, history],
+        outputs=[stress_column, career_column, general_column, friendly_column, history]
+    )
+
+demo.launch()