Update app.py

app.py

@@ -11,6 +11,8 @@ MEDICAL_SYSTEM_PROMPT = """You are DiagnoBot, an AI medical assistant with the f
 2. Offer lifestyle recommendations
 3. Explain medical terminology
 4. Suggest preventive care measures
+5. Based on user's symptoms, preliminary diagnose the disease.
+6. If someone asks you, who created you and who you are, you will simply say you are DiagnoBot and were created by Mahatir Ahmed Tusher, a lone warrior. A ronin.
 You MUST:
 - Maintain HIPAA-compliant confidentiality
 - Clarify when professional consultation is needed
@@ -25,10 +27,9 @@ Do NOT:
 def initialize_messages():
     return [{"role": "system", "content": MEDICAL_SYSTEM_PROMPT}]
 
-messages = initialize_messages()
-
 def get_ai_response(user_input):
-    messages.append({"role": "user", "content": user_input})
+    messages = initialize_messages()
+    messages.append({"role": "user", "content": f"Based on these symptoms: {user_input}, please provide general health information, potential causes to discuss with a doctor, lifestyle recommendations, and dietary suggestions. Format your response with clear sections."})
     
     response = client.chat.completions.create(
         model="llama3-8b-8192",
@@ -39,16 +40,17 @@ def get_ai_response(user_input):
     )
     
     assistant_reply = response.choices[0].message.content
-    messages.append({"role": "assistant", "content": assistant_reply})
     return assistant_reply
 
 # Custom CSS for enhanced styling
 custom_css = """
+body {
+    font-family: 'Times New Roman', serif !important;
+}
 #main-container { 
-    max-width: 800px; 
+    max-width: 900px; 
     margin: auto; 
-    padding: 20px; 
-    font-family: 'Times New Roman', serif;
+    padding: 20px;
 }
 .header {
     text-align: center;
@@ -56,62 +58,100 @@ custom_css = """
     margin-bottom: 20px;
 }
 .logo {
-    max-width: 100%;
+    max-width: 180px;
     height: auto;
+    margin: 0 auto 20px auto;
+    display: block;
+}
+.diagnosis-container {
+    background: #fff;
+    border-radius: 12px;
+    padding: 30px;
+    box-shadow: 0 4px 15px rgba(0, 0, 0, 0.08);
     margin-bottom: 20px;
 }
-.chat-container {
+.result-container {
     background: #f9f9f9;
-    border-radius: 10px;
-    padding: 20px;
-    box-shadow: 0 4px 6px rgba(0, 0, 0, 0.1);
-}
-.user-message { 
-    border-color: #4a90e2 !important; 
-    background: #e3f2fd !important;
+    border-radius: 12px;
+    padding: 25px;
+    box-shadow: 0 4px 10px rgba(0, 0, 0, 0.06);
+    margin-top: 20px;
+    border-left: 5px solid #4a90e2;
 }
-.bot-message { 
-    border-color: #66bb6a !important; 
-    background: #f0fff4 !important;
+.loading-message {
+    text-align: center;
+    font-style: italic;
+    color: #666;
+    padding: 20px;
 }
 .disclaimer {
-    font-size: 0.9em;
-    color: #666;
+    font-size: 0.95em;
+    color: #444;
     margin-top: 20px;
     padding: 15px;
     background: #fff3cd;
-    border-radius: 5px;
+    border-radius: 8px;
+    border-left: 4px solid #ffc107;
 }
 .intro {
     font-size: 1.1em;
     color: #333;
-    margin-bottom: 20px;
+    margin-bottom: 30px;
     text-align: center;
+    line-height: 1.6;
+}
+.submit-btn {
+    background-color: #4a90e2 !important;
+    color: white !important;
+    border-radius: 6px !important;
+    font-weight: bold !important;
+    transition: all 0.3s ease !important;
+}
+.submit-btn:hover {
+    background-color: #3a7bc8 !important;
+    box-shadow: 0 4px 8px rgba(74, 144, 226, 0.3) !important;
+}
+.symptom-input textarea {
+    border: 2px solid #ddd !important;
+    border-radius: 8px !important;
+    transition: border-color 0.3s ease !important;
+}
+.symptom-input textarea:focus {
+    border-color: #4a90e2 !important;
+    box-shadow: 0 0 0 2px rgba(74, 144, 226, 0.2) !important;
 }
 """
 
-def respond(message, chat_history):
-    if not message.strip():
-        return "", chat_history
+# Function to process diagnosis request
+def diagnose(symptoms):
+    if not symptoms.strip():
+        return "Please describe your symptoms to receive a preliminary assessment."
+    
+    # Show loading message
+    yield "Our AI doctor is analyzing your symptoms. Please wait a moment..."
     
-    # Add typing animation
-    start_time = time.time()
+    # Simulate processing time for better UX
+    time.sleep(2)
     
     # Get AI response
-    response = get_ai_response(message)
+    response = get_ai_response(symptoms)
+    
+    # Format the response with better styling
+    formatted_response = f"""
+    ## Preliminary Assessment
     
-    # Simulate typing speed
-    elapsed_time = time.time() - start_time
-    min_type_time = max(1.5, len(response)/50)  # 50 characters per second
+    {response}
     
-    if elapsed_time < min_type_time:
-        time.sleep(min_type_time - elapsed_time)
+    ---
     
-    chat_history.append((message, response))
-    return "", chat_history
+    *Remember: This is preliminary information only. Always consult with a healthcare professional for proper diagnosis and treatment.*
+    """
+    
+    yield formatted_response
 
+# Main UI setup
 with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
-    # Logo at the top
+    # Logo and introduction
     gr.HTML("""<div class="header">
                <img src="logo.png" alt="DiagnoBot Logo" class="logo">
              </div>""")
@@ -125,66 +165,53 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Soft()) as demo:
     </div>
     """)
     
-    # Chat interface
-    with gr.Row():
-        with gr.Column(scale=2):
-            chatbot = gr.Chatbot(
-                label="Consultation History",
-                bubble_full_width=False,
-                avatar_images=(
-                    "https://cdn-icons-png.flaticon.com/512/1077/1077012.png",  # User avatar
-                    "https://cdn-icons-png.flaticon.com/512/4712/4712035.png"   # Bot avatar
-                ),
-                height=500
-            )
-            
-            with gr.Row():
-                user_input = gr.Textbox(
-                    placeholder="Describe your symptoms...",
-                    show_label=False,
-                    container=False,
-                    autofocus=True
-                )
-                submit_btn = gr.Button("Send", variant="primary")
-            
-            # Example symptoms
-            gr.Examples(
-                examples=[
-                    ["Headache and fever"],
-                    ["I have slept enough yet I am having a bad headache"],
-                    ["Chest pain and shortness of breath"],
-                    ["Fatigue and dizziness"],
-                    ["Abdominal pain and nausea"]
-                ],
-                inputs=user_input
-            )
-            
-            # Disclaimer at the bottom
-            gr.Markdown("""
-            <div class="disclaimer">
-            **Disclaimer:** This AI provides general health information and should not be used for emergency situations or as a substitute for professional medical advice. Always consult a qualified healthcare provider for personal health concerns.
-            </div>
-            """)
-            
-            # Tools section
-            with gr.Accordion("Consultation Tools", open=False):
-                with gr.Row():
-                    gr.Button("New Conversation", variant="secondary")
-                    gr.Button("Export Chat", variant="secondary")
-                    gr.Button("Emergency Resources", variant="stop")
+    # Main diagnosis container
+    with gr.Column(elem_classes="diagnosis-container"):
+        gr.Markdown("### Describe Your Symptoms")
+        
+        # Symptom input
+        symptoms_input = gr.Textbox(
+            placeholder="Please describe your symptoms in detail...",
+            lines=5,
+            label="",
+            elem_classes="symptom-input"
+        )
+        
+        diagnose_btn = gr.Button("Get Preliminary Assessment", variant="primary", elem_classes="submit-btn")
+        
+        # Example symptoms
+        gr.Examples(
+            examples=[
+                ["Headache and fever for the past two days"],
+                ["I have slept enough yet I am having a bad headache accompanied by sensitivity to light"],
+                ["Chest pain and shortness of breath after minimal exertion"],
+                ["Persistent fatigue and dizziness, especially when standing up quickly"],
+                ["Abdominal pain in the lower right side and nausea that worsens after eating"]
+            ],
+            inputs=symptoms_input,
+            label="Common Symptom Examples"
+        )
+        
+        # Results container
+        diagnosis_output = gr.Markdown(elem_classes="result-container")
+    
+    # Disclaimer at the bottom
+    gr.Markdown("""
+    <div class="disclaimer">
+    <strong>Important Disclaimer:</strong> This AI provides general health information and preliminary insights based on described symptoms. It should NOT be used for emergency situations or as a substitute for professional medical advice. The information provided is not a diagnosis. Always consult a qualified healthcare provider for personal health concerns. If you're experiencing severe symptoms, please seek immediate medical attention.
+    </div>
+    """)
     
     # Interaction logic
-    submit_btn.click(
-        respond,
-        [user_input, chatbot],
-        [user_input, chatbot],
-        queue=False
+    diagnose_btn.click(
+        diagnose,
+        inputs=[symptoms_input],
+        outputs=[diagnosis_output]
     )
-    user_input.submit(
-        respond,
-        [user_input, chatbot],
-        [user_input, chatbot],
-        queue=False
+    symptoms_input.submit(
+        diagnose,
+        inputs=[symptoms_input],
+        outputs=[diagnosis_output]
     )
 
 demo.launch(share=True)