Update app.py

app.py

@@ -24,18 +24,22 @@ def respond(
     temperature,
     top_p,
 ):
-    """Generate response from the medical chatbot"""
+    """
+    Generate response from the medical chatbot
+    """
     # Format the prompt with conversation history
     conversation = system_message + "\n\n"
     
     # Add chat history
     for entry in history:
         if isinstance(entry, dict):
+            # New format: messages with 'role' and 'content'
             if entry.get('role') == 'user':
                 conversation += f"User: {entry['content']}\n"
             elif entry.get('role') == 'assistant':
                 conversation += f"Bot: {entry['content']}\n"
         else:
+            # Old format: tuples
             user_msg, bot_msg = entry
             conversation += f"User: {user_msg}\nBot: {bot_msg}\n"
     
@@ -43,20 +47,15 @@ def respond(
     conversation += f"User: {message}\nBot:"
     
     # Tokenize and generate
-    inputs = tokenizer(
-        conversation, 
-        return_tensors="pt", 
-        truncation=True, 
-        max_length=1024
-    ).to(device)
+    inputs = tokenizer(conversation, return_tensors="pt", truncation=True, max_length=1024).to(device)
     
     with torch.no_grad():
         outputs = model.generate(
             **inputs,
-            max_new_tokens=int(max_tokens),
+            max_new_tokens=max_tokens,
             do_sample=True,
-            temperature=float(temperature),
-            top_p=float(top_p),
+            temperature=temperature,
+            top_p=top_p,
             pad_token_id=tokenizer.eos_token_id,
             eos_token_id=tokenizer.eos_token_id,
         )
@@ -67,6 +66,7 @@ def respond(
     # Extract only the bot's response
     if "Bot:" in full_response:
         response = full_response.split("Bot:")[-1].strip()
+        # Remove any trailing "User:" if present
         if "User:" in response:
             response = response.split("User:")[0].strip()
     else:
@@ -75,170 +75,91 @@ def respond(
     return response
 
 
-# Create custom CSS for larger, more visible send button
+# Add CSS to make Send button more visible
 custom_css = """
-#send_btn {
+button[type="submit"] {
+    min-width: 100px !important;
+    padding: 10px 20px !important;
+    font-size: 14px !important;
+    font-weight: 600 !important;
     background: linear-gradient(135deg, #667eea 0%, #764ba2 100%) !important;
+    border: 2px solid #667eea !important;
     color: white !important;
-    font-size: 16px !important;
-    font-weight: bold !important;
-    padding: 12px 30px !important;
-    border-radius: 8px !important;
-    border: none !important;
+    border-radius: 6px !important;
     cursor: pointer !important;
-    box-shadow: 0 4px 15px rgba(102, 126, 234, 0.4) !important;
-    transition: all 0.3s ease !important;
-    height: 50px !important;
-    min-width: 120px !important;
+    box-shadow: 0 4px 12px rgba(102, 126, 234, 0.4) !important;
+    transition: all 0.2s ease !important;
 }
 
-#send_btn:hover {
+button[type="submit"]:hover {
     background: linear-gradient(135deg, #764ba2 0%, #667eea 100%) !important;
-    box-shadow: 0 6px 20px rgba(102, 126, 234, 0.6) !important;
+    box-shadow: 0 6px 16px rgba(102, 126, 234, 0.6) !important;
     transform: translateY(-2px) !important;
 }
-
-#send_btn:active {
-    transform: translateY(0) !important;
-}
-
-/* Make input textbox larger */
-#message_input textarea {
-    font-size: 15px !important;
-    padding: 15px !important;
-    min-height: 60px !important;
-}
-
-/* Improve overall layout */
-.gr-chatbot {
-    min-height: 600px !important;
-}
 """
 
-# Create Gradio ChatInterface with visible Send button
-with gr.Blocks(css=custom_css) as demo:
-    gr.HTML("""
-        <div style='text-align: center; padding: 20px;'>
-            <h1>🏥 Medical Symptom Chatbot</h1>
-            <p style='color: #666; font-size: 16px;'>
-                Ask questions about symptoms, diseases, and medical conditions
-            </p>
-            <p style='color: #e74c3c; font-size: 14px; font-weight: bold;'>
-                ⚠️ For informational purposes only - Always consult healthcare professionals
-            </p>
-        </div>
-    """)
-    
-    with gr.Row():
-        chatbot = gr.Chatbot(
-            label="Chat History",
-            height=500,
-            show_copy_button=True,
-            bubble_full_width=False
-        )
-    
-    with gr.Row():
-        with gr.Column(scale=8):
-            message_input = gr.Textbox(
-                elem_id="message_input",
-                placeholder="Ask me about your symptoms or medical questions...",
-                label="Your Message",
-                lines=2,
-                show_label=True,
-            )
-        with gr.Column(scale=2):
-            send_btn = gr.Button(
-                "📤 SEND",
-                elem_id="send_btn",
-                variant="primary",
-                size="lg",
-                scale=1,
-            )
-    
-    with gr.Row():
-        with gr.Accordion("⚙️ Advanced Settings", open=False):
-            with gr.Row():
-                system_message = gr.Textbox(
-                    value="You are a helpful medical chatbot that provides information about symptoms and diseases. Always recommend consulting a healthcare professional for serious conditions.",
-                    label="System Message",
-                    lines=2,
-                )
-            
-            with gr.Row():
-                max_tokens = gr.Slider(
-                    minimum=50,
-                    maximum=300,
-                    value=150,
-                    step=10,
-                    label="Max Tokens",
-                    info="Maximum length of the response"
-                )
-                temperature = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.0,
-                    value=0.7,
-                    step=0.1,
-                    label="Temperature",
-                    info="Higher = more creative"
-                )
-                top_p = gr.Slider(
-                    minimum=0.1,
-                    maximum=1.0,
-                    value=0.9,
-                    step=0.05,
-                    label="Top-p",
-                    info="Controls diversity"
-                )
-    
-    # Examples
-    gr.Examples(
-        examples=[
-            ["I have fever and cough. What could this be?"],
-            ["What are the symptoms of diabetes?"],
-            ["What is hypertension?"],
-            ["I have a headache and nausea. What should I do?"],
-            ["What are the precautions for common cold?"],
-        ],
-        inputs=message_input,
-        label="📝 Example Questions"
-    )
-    
-    # Chat history state
-    chat_history = gr.State([])
-    
-    # Handle send button click
-    def chat(message, history, system_msg, tokens, temp, top_p_val):
-        if not message.strip():
-            return history, ""
-        
-        response = respond(message, history, system_msg, tokens, temp, top_p_val)
-        history.append({"role": "user", "content": message})
-        history.append({"role": "assistant", "content": response})
-        
-        return history, ""
-    
-    # Connect send button
-    send_btn.click(
-        fn=chat,
-        inputs=[message_input, chat_history, system_message, max_tokens, temperature, top_p],
-        outputs=[chatbot, message_input]
-    ).then(
-        lambda: None,
-        outputs=chat_history
-    )
-    
-    # Allow Enter key to send
-    message_input.submit(
-        fn=chat,
-        inputs=[message_input, chat_history, system_message, max_tokens, temperature, top_p],
-        outputs=[chatbot, message_input]
-    )
-
+# Create Gradio ChatInterface
+chatbot = gr.ChatInterface(
+    respond,
+    type="messages",  # Use new message format
+    chatbot=gr.Chatbot(
+        height=500,
+    ),
+    textbox=gr.Textbox(
+        placeholder="Ask me about your symptoms or medical questions...",
+        container=True,  # keep the input container visible so the send button renders
+        scale=7
+    ),
+    title="🏥 Medical Symptom Chatbot",
+    description="Ask questions about symptoms, diseases, and medical conditions. This bot is trained on medical Q&A data. For informational purposes only - always consult healthcare professionals.",
+    theme="soft",
+    examples=[
+        ["I have fever and cough. What could this be?"],
+        ["What are the symptoms of diabetes?"],
+        ["What is hypertension?"],
+        ["I have a headache and nausea. What should I do?"],
+        ["What are the precautions for common cold?"],
+    ],
+    cache_examples=False,
+    additional_inputs=[
+        gr.Textbox(
+            value="You are a helpful medical chatbot that provides information about symptoms and diseases. Always recommend consulting a healthcare professional for serious conditions.",
+            label="System Message",
+            lines=3
+        ),
+        gr.Slider(
+            minimum=50,
+            maximum=300,
+            value=150,
+            step=10,
+            label="Max Tokens",
+            info="Maximum length of the response"
+        ),
+        gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.7,
+            step=0.1,
+            label="Temperature",
+            info="Higher = more creative, Lower = more focused"
+        ),
+        gr.Slider(
+            minimum=0.1,
+            maximum=1.0,
+            value=0.9,
+            step=0.05,
+            label="Top-p (Nucleus Sampling)",
+            info="Controls diversity of responses"
+        ),
+    ],
+    submit_btn="Send",
+    css=custom_css,
+)
 
+# Launch the app
 if __name__ == "__main__":
-    demo.launch(
-        share=False,
-        server_name="0.0.0.0",
+    chatbot.launch(
+        share=False,  # Set to True to create a public link
+        server_name="0.0.0.0",  # Makes it accessible externally
         server_port=7860,
-        show_error=True,
     )