Update app.py

app.py

@@ -1,147 +1,86 @@
 from langchain_google_genai import GoogleGenerativeAI
 import streamlit as st
-import re
-from datetime import datetime
 
-# --- Configuration ---
-MODEL_CONFIG = {
-    "model": "gemini-pro",
-    "temperature": 0.7,
-    "top_p": 0.85,
-    "max_tokens": 1024,
-    "safety_settings": {
-        "HARASSMENT":"BLOCK_ONLY_HIGH",
-        "HATE_SPEECH":"BLOCK_ONLY_HIGH",
-        "SEXUAL":"BLOCK_ONLY_HIGH",
-        "DANGEROUS":"BLOCK_ONLY_HIGH"
-    }
-}
-
-EMERGENCY_CONTACTS = {
-    "Hospital Address": "John Smith Hospital, 123 Health St.",
-    "Ambulance": "911 (US) / 112 (EU)",
-    "24/7 Support": "+1-800-123-4567"
-}
-
-# --- Security & Sanitization ---
-def sanitize_input(text):
-    """Remove potentially harmful patterns"""
-    return re.sub(r"[<>{}[\]~`]", "", text[:2000])
-
-# --- AI Response Validation ---
-def validate_medical_response(text):
-    """Basic safety checks for AI responses"""
-    blacklist = ["take your own life", "kill yourself", "hate you"]
-    return not any(phrase in text.lower() for phrase in blacklist)
+# --- LLM Initialization ---
+llm = GoogleGenerativeAI(
+    model="gemini-1.5-flash",
+    google_api_key=GEMINI_API_KEY
+    temperature= 0.7,
+    top_p= 0.85,
+    max_tokens= 1024,
+    stop_sequence = ["take your own life", "kill yourself", "hate you"]
+)
+
+# --- instruction ---
+instructions = "Answer the user's question politely and provide accurate information."
+
+# --- Context & Input Data ---
+context =(
+    "You are DoctorX, an AI medical assistant. Follow these rules STRICTLY:"
+    "1. Never diagnose - suggest POSSIBLE conditions"
+    "2. Always recommend consulting a human doctor"
+    "3. Prioritize patient safety over all else"
+    "4. Maintain professional, empathetic tone"
+    "5. Disclose your AI nature clearly"
+    "6. For emergencies, direct to sidebar contacts"
+)
+
+input_data = (
+    "Hospital Address:John Smith Hospital, 123 Health St."
+    "Ambulance:911 (US) / 112 (EU)"
+    "24/7 Support:+1-800-123-4567"
+)
 
 # --- Streamlit App ---
 def main():
     st.set_page_config(
-        page_title="DoctorX: AI Health Assistant",
+        page_title="DoctorX",
         page_icon="🩺",
-        layout="centered",
-        initial_sidebar_state="collapsed"
+        layout="wide",
+        initial_sidebar_state="expanded"
     )
 
-    # Initialize session state
+    # Initialize session state for chat history
     if "history" not in st.session_state:
         st.session_state.history = []
-    
-    # --- Sidebar with Emergency Info ---
+
+    # Sidebar for emergency contacts
     with st.sidebar:
-        st.header("🆘 Emergency Contacts")
-        for key, value in EMERGENCY_CONTACTS.items():
+        st.header("🚨 Emergency Contacts")
+        for key, value in input_data.items():
             st.subheader(key)
             st.caption(value)
         st.divider()
-        st.caption("⚠️ This is not a substitute for emergency medical care")
+        st.caption("⚠️ This is not a substitute for emergency medical care.")
 
-    # --- Main Interface ---
-    st.title("🩺 DoctorX Health Assistant")
-    st.caption("AI-powered preliminary health guidance - Always consult a human doctor for final diagnosis")
+    # Main interface
+    st.title("🩺 DoctorX")
+    st.caption("AI-powered preliminary health guidance - Always consult a human doctor for final diagnosis.")
 
     # Display chat history
     for role, message in st.session_state.history:
         with st.chat_message(role):
             st.markdown(message)
 
-    # User input with enhanced options
-    with st.form("chat_input"):
-        col1, col2 = st.columns([4, 1])
-        with col1:
-            user_input = st.text_input(
-                "Describe your symptoms or ask a question:",
-                placeholder="E.g., I've had a headache for 3 days...",
-                max_chars=500
-            )
-        with col2:
-            if st.form_submit_button("Send", use_container_width=True):
-                user_input = sanitize_input(user_input.strip())
-                
-                if user_input.lower() in ["emergency", "911", "112"]:
-                    st.session_state.history.append(("user", user_input))
-                    st.session_state.history.append(("bot", "🚨 Please use the emergency contacts in the sidebar immediately!"))
-                    st.rerun()
-
-                elif user_input:
-                    process_user_input(user_input)
-
-    # Quick symptom buttons
-    common_symptoms = ["Fever", "Headache", "Cough", "Rash", "Nausea"]
-    cols = st.columns(len(common_symptoms))
-    for col, symptom in zip(cols, common_symptoms):
-        if col.button(symptom):
-            process_user_input(f"I'm experiencing {symptom.lower()}")
+    # User input
+    user_query = st.text_input("Describe your symptoms or ask a question:", placeholder="e.g., I have a headache")
+    if user_query:
+        st.session_state.history.append(("user", user_query))
+        with st.chat_message("bot"):
+            with st.spinner("Analyzing your input..."):
+                response = llm.invoke(user_query)
+                st.markdown(response)
+                st.session_state.history.append(("bot", response))
 
-    # Quick response buttons for common queries
+    # Quick response buttons
     common_queries = ["What are the symptoms of epilepsy?", "How to manage stress?"]
     cols = st.columns(len(common_queries))
     for col, query in zip(cols, common_queries):
         if col.button(query):
-            process_user_input(query)
-
-def process_user_input(user_input):
-    """Handle user input and generate AI response"""
-    llm = GoogleGenerativeAI(google_api_key=st.secrets["GOOGLE_API_KEY"], **MODEL_CONFIG)
-    
-    # Build medical context
-    system_prompt = f"""
-    You are DoctorX, an AI medical assistant. Follow these rules STRICTLY:
-    1. Never diagnose - suggest POSSIBLE conditions
-    2. Always recommend consulting a human doctor
-    3. Prioritize patient safety over all else
-    4. Maintain professional, empathetic tone
-    5. Disclose your AI nature clearly
-    6. For emergencies, direct to sidebar contacts
-    
-    Current Date: {datetime.now().strftime("%Y-%m-%d")}
-    User Input: {user_input}
-    """
-    
-    try:
-        with st.chat_message("user"):
-            st.markdown(user_input)
-            
-        with st.chat_message("bot"):
-            with st.spinner("🔍 Analyzing symptoms..."):
-                response = llm.invoke(system_prompt)
-                
-                if not validate_medical_response(response):
-                    raise ValueError("Response validation failed")
-                
-                # Format medical information
-                response = response.replace("**Possible Condition:**", "\n🩺 **Possible Condition:**")
-                response = response.replace("**Recommendation:**", "\n📋 **Recommendation:**")
-                
-                st.markdown(response)
-                st.caption("ℹ️ AI-generated suggestion - Verify with healthcare professional")
-
-            st.session_state.history.append(("user", user_input))
+            st.session_state.history.append(("user", query))
+            response = llm.invoke(query)
             st.session_state.history.append(("bot", response))
-
-    except Exception as e:
-        st.error(f"⚠️ System Error: {str(e)}")
-        st.session_state.history.append(("bot", "🚨 Error processing request. Please try again later."))
+            st.markdown(response)
 
 if __name__ == "__main__":
     main()