Update app.py

app.py

@@ -8,13 +8,11 @@ openai.api_base = "https://api.groq.com/openai/v1"
 
 # Function to get response from GROQ API
 def get_groq_response(message, category):
-    # Define system message based on category
     system_messages = {
         "Study Tips": "Provide effective study strategies, time management advice, and tips for staying organized and focused.",
         "Exam Preparation": "Offer tips for preparing for exams, managing anxiety, and optimizing performance on test day.",
         "Project Guidance": "Provide advice on how to tackle academic projects, group assignments, and presentations effectively.",
         "Stress Management": "Offer calming techniques and advice to handle stress effectively.",
-        # (Continue with all other categories...)
         "Friendly Buddy": "Respond as a supportive and fun friend. Be informal, cheerful, and light-hearted."
     }
     system_message = system_messages.get(category, "Respond appropriately to the user's input.")
@@ -37,7 +35,7 @@ def chatbot(user_input, category, history=[]):
     history.append((f"You: {user_input}", f"Bot: {bot_response}"))
     return history, history
 
-# Gradio Interface with grouped categories
+# Categories grouped into main and subcategories
 categories = {
     "Academic Support": [
         "Study Tips", "Exam Preparation", "Project Guidance", "Time Management", "Building Confidence"
@@ -57,12 +55,8 @@ categories = {
     ]
 }
 
-def get_subcategories(main_category):
-    return categories.get(main_category, [])
-
-with gr.Blocks(css="""
-/* Add custom CSS here */
-""") as chat_interface:
+# Gradio Interface with grouped categories
+with gr.Blocks() as chat_interface:
     with gr.Row():
         gr.Markdown("<h1 style='text-align:center;'>🌟 Vibrant Personal Assistant Chatbot 🌈</h1>")
     
@@ -77,11 +71,14 @@ with gr.Blocks(css="""
         )
         sub_category = gr.Dropdown(
             label="Subcategory",
-            choices=get_subcategories("Academic Support")
+            choices=categories["Academic Support"],
+            value="Study Tips"
         )
 
-    def update_subcategories(main_category):
-        return gr.Dropdown.update(choices=get_subcategories(main_category), value=None)
+    def update_subcategories(selected_main_category):
+        """Update the subcategory dropdown based on the main category."""
+        new_subcategories = categories.get(selected_main_category, [])
+        return gr.Dropdown.update(choices=new_subcategories, value=new_subcategories[0] if new_subcategories else None)
 
     main_category.change(update_subcategories, inputs=main_category, outputs=sub_category)
 
@@ -92,11 +89,10 @@ with gr.Blocks(css="""
     with gr.Row():
         chatbot_output = gr.Chatbot(label="Chat History")
 
-    # Add functionality to handle interactions
-    def handle_chat(user_input, category, history):
+    def handle_chat(user_input, sub_category, history):
         if not user_input.strip():
             return history, history
-        updated_history, _ = chatbot(user_input, category, history)
+        updated_history, _ = chatbot(user_input, sub_category, history)
         return updated_history, updated_history
 
     send_button.click(