Update Gradio_UI.py

Gradio_UI.py

@@ -194,11 +194,35 @@ class GradioUI:
 
         messages.append(gr.ChatMessage(role="user", content=prompt))
         yield messages
+
+        # Add loading message
+        loading_msg = gr.ChatMessage(
+            role="assistant",
+            content="Thinking...🤔👩🏽‍⚕️👨🏽‍⚕️",
+            metadata={"class": "loading-message"}
+        )
+        messages.append(loading_msg)
+        yield messages
+
+        # Process agent response
         for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
+            # Remove loading message before adding real responses
+            if len(messages) > 0 and messages[-1].metadata.get("class") == "loading-message":
+                messages.pop()
             messages.append(msg)
             yield messages
+
+        # Ensure loading message is removed at the end
+        if len(messages) > 0 and messages[-1].metadata.get("class") == "loading-message":
+            messages.pop()
         yield messages
 
+        
+        # for msg in stream_to_gradio(self.agent, task=prompt, reset_agent_memory=False):
+        #     messages.append(msg)
+        #     yield messages
+        # yield messages
+
     def upload_file(
         self,
         file,
@@ -261,7 +285,26 @@ class GradioUI:
     def launch(self, **kwargs):
         import gradio as gr
 
-        with gr.Blocks(fill_height=True) as demo:
+        with gr.Blocks(
+            title="Health Agent",
+            fill_height=True,
+            css="""
+                .loading-message {
+                    animation: pulse 2s cubic-bezier(.4,0,.6,1) infinite;
+                }
+                @keyframes pulse {
+                    0%, 100% { opacity: 1; }
+                    50% { opacity: 0.5; }
+                }
+            """
+        ) as demo:
+            gr.Markdown(
+                """
+                # NELA - AI-Powered Health Advisor  
+                I am NELA, your AI-powered health advisor. I can help you track your health, provide symptom analysis, suggest possible diagnoses, and offer wellness recommendations.  
+                Ask me about symptoms, treatments, medications, or how to maintain a healthy lifestyle!
+                """
+            )
             stored_messages = gr.State([])
             file_uploads_log = gr.State([])
             chatbot = gr.Chatbot(
@@ -274,22 +317,69 @@ class GradioUI:
                 resizeable=True,
                 scale=1,
             )
-            # If an upload folder is provided, enable the upload feature
-            if self.file_upload_folder is not None:
-                upload_file = gr.File(label="Upload a file")
-                upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
-                upload_file.change(
-                    self.upload_file,
-                    [upload_file, file_uploads_log],
-                    [upload_status, file_uploads_log],
-                )
-            text_input = gr.Textbox(lines=1, label="Chat Message")
+            text_input = gr.Textbox(
+                label="Ask me anything about your health",
+                placeholder="Type your question here...",
+            )
+            submit_button = gr.Button("Send")
+            gr.Examples(
+            examples=[
+                "What could be the possible causes of persistent headaches?",
+                "Suggest some home remedies for managing cold and flu symptoms.",
+                "What are the early signs of diabetes?",
+                "How can I lower my blood pressure naturally?",
+                "What should I do if I experience sudden chest pain?",
+                "Which foods are best for boosting my immune system?",
+                "How can I track my daily health using smart devices?",
+                "What are common symptoms of dehydration and how can I prevent it?",
+                "Explain the importance of regular health check-ups.",
+                "What should I include in a balanced diet for better heart health?"
+            ],
+                inputs=text_input,
+                label="Example Questions"
+            )
             text_input.submit(
                 self.log_user_message,
                 [text_input, file_uploads_log],
                 [stored_messages, text_input],
             ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
 
+            submit_button.click(
+                self.log_user_message,
+                [text_input, file_uploads_log],
+                [stored_messages, text_input],
+            ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
+
+        
+        # with gr.Blocks(fill_height=True) as demo:
+        #     stored_messages = gr.State([])
+        #     file_uploads_log = gr.State([])
+        #     chatbot = gr.Chatbot(
+        #         label="Agent",
+        #         type="messages",
+        #         avatar_images=(
+        #             None,
+        #             "https://huggingface.co/datasets/agents-course/course-images/resolve/main/en/communication/Alfred.png",
+        #         ),
+        #         resizeable=True,
+        #         scale=1,
+        #     )
+        #     # If an upload folder is provided, enable the upload feature
+        #     if self.file_upload_folder is not None:
+        #         upload_file = gr.File(label="Upload a file")
+        #         upload_status = gr.Textbox(label="Upload Status", interactive=False, visible=False)
+        #         upload_file.change(
+        #             self.upload_file,
+        #             [upload_file, file_uploads_log],
+        #             [upload_status, file_uploads_log],
+        #         )
+        #     text_input = gr.Textbox(lines=1, label="Chat Message")
+        #     text_input.submit(
+        #         self.log_user_message,
+        #         [text_input, file_uploads_log],
+        #         [stored_messages, text_input],
+        #     ).then(self.interact_with_agent, [stored_messages, chatbot], [chatbot])
+
         demo.launch(debug=True, share=True, **kwargs)