Update app.py

app.py

@@ -143,7 +143,7 @@ with gr.Blocks() as demo:
                     value="Moderately active"
                 )
 
-            chatbot = gr.Chatbot(label="Chat with FIT.AI")
+            chatbot = gr.Chatbot(label="Chat with FIT.AI", type="messages")
             text_input = gr.Textbox(placeholder="Type your question here...", label="Your Question")
             submit_button = gr.Button("Submit")
             clear_button = gr.Button("Clear Chat")
@@ -162,14 +162,15 @@ with gr.Blocks() as demo:
                 bmi_path = visualize_bmi(bmi)
                 calories = calculate_calories(user_details["age"], user_details["weight"], user_details["height"], user_details["activity_level"], user_details["gender"])
                 response = run_graph(message, history, user_details)
-                history.append(("User", message))
-                history.append(("FIT.AI", response + f"\nYour BMI is {bmi:.2f}, considered {status}.\nDaily calorie needs: {calories} kcal."))
-                return history, "", bmi_path
+                history.append({"role": "user", "content": message})
+                history.append({"role": "assistant", "content": response + f"\nYour BMI is {bmi:.2f}, considered {status}.\nDaily calorie needs: {calories} kcal.", "image": bmi_path})
+                return history, ""
             else:
-                return history, "Error in calculation.", ""
+                history.append({"role": "assistant", "content": "Error in calculation."})
+                return history, ""
 
-        submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input, chatbot])
-        clear_button.click(lambda: ([], "", ""), inputs=None, outputs=[chatbot, text_input, chatbot])
+        submit_button.click(submit_message, inputs=[text_input, chatbot], outputs=[chatbot, text_input])
+        clear_button.click(lambda: ([], ""), inputs=None, outputs=[chatbot, text_input])
 
     with gr.Tab("BMI and Calorie Calculator"):
         with gr.Row():
@@ -206,4 +207,4 @@ with gr.Blocks() as demo:
             outputs=[bmi_output, calorie_output, bmi_chart]
         )
 
-demo.launch(share=True)
+demo.launch(share=True)