Update app.py

app.py

@@ -72,6 +72,28 @@ def generate_empathetic_response(user_input, retrieved_question):
         return response.choices[0].message.content
     except Exception as e:
         return "I'm sorry, I couldn't process your request."
+# ✅ Function to detect disorders
+def detect_disorders(chat_history):
+    """Detect psychiatric disorders from full chat history."""
+    full_chat_text = " ".join(chat_history)
+    text_embedding = similarity_model.encode([full_chat_text], convert_to_numpy=True)
+    distances, indices = index.search(text_embedding, 3)
+
+    if indices[0][0] == -1:
+        return ["No matching disorder found."]
+
+    disorders = [recommendations_df["Disorder"].iloc[i] for i in indices[0]]
+    return disorders
+
+# ✅ Function to get treatment recommendations
+def get_treatment(detected_disorders):
+    """Retrieve treatment recommendations based on detected disorders."""
+    treatments = {
+        disorder: recommendations_df[recommendations_df["Disorder"] == disorder]["Treatment Recommendation"].values[0]
+        for disorder in detected_disorders if disorder in recommendations_df["Disorder"].values
+    }
+    return treatments
+
 
 # ✅ Streamlit UI Setup
 st.title("🧠 MindSpark AI Psychiatric Assistant")
@@ -97,3 +119,21 @@ if st.button("Summarize Chat"):
     summary = summarization_tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     st.subheader("Chat Summary")
     st.write(summary)
+if st.button("Detect Disorders"):
+    if chat_history.strip():
+        disorders = detect_disorders(chat_history.split("\n"))
+        st.write("**Detected Disorders:**")
+        for disorder in disorders:
+            st.write(f"- {disorder}")
+    else:
+        st.error("❌ Please enter chat history.")
+
+if st.button("Get Treatment Recommendations"):
+    if chat_history.strip():
+        detected_disorders = detect_disorders(chat_history.split("\n"))
+        treatments = get_treatment(detected_disorders)
+        st.write("**Treatment Recommendations:**")
+        for disorder, treatment in treatments.items():
+            st.write(f"**{disorder}:** {treatment}")
+    else:
+        st.error("❌ Please enter chat history.")