Update app.py

app.py

@@ -1,48 +1,39 @@
-# app.py
+import os
 import gradio as gr
 import spaces
 from agent import generate_response
 
+user_preferences = {
+    "diet": [],
+    "goal": "",
+    "weeks": 1
+}
+
 @spaces.GPU
-def main():
-    with gr.Blocks(title="Meal Plan Chatbot") as demo:
-        gr.Markdown("# 🍎 MealBot: Your AI Nutritionist")
-
-        with gr.Row():
-            diet_type = gr.CheckboxGroup(
-                label="Diet Type",
-                choices=["Vegan", "Keto", "Vegetarian", "Paleo", "High-Protein"],
-                value=["Vegetarian"]
-            )
-            goal = gr.CheckboxGroup(
-                label="Your Goal",
-                choices=["Weight Loss", "Muscle Gain", "Maintenance"],
-                value=["Weight Loss"]
-            )
-            duration = gr.CheckboxGroup(
-                label="Duration",
-                choices=["1 week", "4 weeks", "8 weeks"],
-                value=["4 weeks"]
-            )
-
-        chatbot = gr.Chatbot()
-        msg = gr.Textbox(placeholder="Ask me anything about meal plans...")
-        clear = gr.Button("Clear")
-
-        def user_respond(message, chat_history):
-            prefs = {
-                "diet": diet_type.value,
-                "goal": goal.value,
-                "duration": duration.value
-            }
-            response = generate_response(message, prefs)
-            chat_history.append((message, response))
-            return "", chat_history
-
-        msg.submit(user_respond, [msg, chatbot], [msg, chatbot])
-        clear.click(lambda: None, None, chatbot, queue=False)
-
-    demo.launch()
-
-if __name__ == "__main__":
-    main()
+def process_input(message, history):
+    return generate_response(message, history, user_preferences)
+
+def update_preferences(diet, goal, weeks):
+    user_preferences["diet"] = diet
+    user_preferences["goal"] = goal
+    user_preferences["weeks"] = weeks
+    return gr.update(visible=True)
+
+with gr.Blocks() as demo:
+    gr.Markdown("## 🥗 Smart Meal Plan Assistant")
+
+    with gr.Row():
+        diet = gr.CheckboxGroup(["Low-carb", "High-protein", "Vegetarian", "Vegan", "Keto"], label="Diet Preferences")
+        goal = gr.Radio(["Weight Loss", "Muscle Gain", "Maintenance"], label="Health Goal")
+        weeks = gr.Slider(1, 12, value=1, step=1, label="Plan Duration (weeks)")
+
+    confirm_btn = gr.Button("Start Chat")
+    chatbot = gr.Chatbot(visible=False)
+    msg = gr.Textbox(visible=False)
+    clear = gr.Button("Clear", visible=False)
+
+    confirm_btn.click(update_preferences, inputs=[diet, goal, weeks], outputs=[chatbot])
+    msg.submit(process_input, [msg, chatbot], chatbot)
+    clear.click(lambda: None, None, chatbot)
+
+demo.launch()