Spaces:
Running
on
Zero
Running
on
Zero
| import gradio as gr | |
| from .handlers import respond, reset_chat | |
| def user_message_handler(user_message, chat_history): | |
| # Immediately show the user's message | |
| chat_history = chat_history + [(user_message, None)] | |
| return "", chat_history | |
| def bot_response_handler(chat_history): | |
| # Get the last user message | |
| user_message = chat_history[-1][0] | |
| # Get the bot's response | |
| _, updated_history = respond(user_message, chat_history[:-1]) | |
| # Replace the last (user, None) with (user, bot_response) | |
| chat_history[-1] = updated_history[-1] | |
| return chat_history | |
| def build_interface(): | |
| with gr.Blocks(theme="soft") as demo: | |
| gr.Markdown("# 🏥 Dr Bavana's Medical Assistant ") | |
| gr.Markdown("By Techindia Solutions AI") | |
| with gr.Row(): | |
| with gr.Column(scale=4): | |
| chatbot = gr.Chatbot(height=500) | |
| msg = gr.Textbox( | |
| placeholder="Tell me about your symptoms or health concerns...", | |
| label="Your Message" | |
| ) | |
| send_btn = gr.Button("Send", elem_id="send-btn") | |
| with gr.Column(scale=1): | |
| reset_btn = gr.Button("🔄 Start New Consultation", variant="secondary") | |
| gr.Markdown("**Memory Features:**\n- Tracks symptoms & timeline\n- Remembers medications & allergies\n- Maintains conversation context\n- Provides comprehensive summaries") | |
| gr.Examples( | |
| examples=[ | |
| "I have a persistent cough and sore throat for 3 days", | |
| "I've been having severe headaches and feel dizzy", | |
| "My stomach hurts and I feel nauseous after eating" | |
| ], | |
| inputs=msg | |
| ) | |
| # Step 1: Show user message instantly | |
| msg.submit(user_message_handler, [msg, chatbot], [msg, chatbot]).then( | |
| bot_response_handler, chatbot, chatbot | |
| ) | |
| send_btn.click(user_message_handler, [msg, chatbot], [msg, chatbot]).then( | |
| bot_response_handler, chatbot, chatbot | |
| ) | |
| reset_btn.click(reset_chat, [], [chatbot, msg]) | |
| return demo |