Update app.py

app.py

@@ -78,7 +78,40 @@ def interact_with_together_api(messages, model_link, diagnostic=False):
 
 # Function to diagnose mental health issue
 def diagnose_mental_health(messages):
-    return interact_with_together_api(messages, model_link, diagnostic=True)
+    diagnostic_prompt = "Analyze the following conversation and predict the mental issue the user might be suffering from:"
+    diagnostic_messages = [{"role": "system", "content": "You are an AI model specialized in mental health diagnosis."},
+                           {"role": "user", "content": diagnostic_prompt}]
+    for message in messages:
+        if message[0] == "user":
+            diagnostic_messages.append({"role": "user", "content": message[1]})
+    
+    url = "https://api.together.xyz/v1/chat/completions"
+    payload = {
+        "model": model_link,
+        "temperature": 0.7,
+        "top_p": 0.9,
+        "top_k": 50,
+        "repetition_penalty": 1.2,
+        "n": 1,
+        "messages": diagnostic_messages,
+    }
+
+    TOGETHER_API_KEY = os.getenv('TOGETHER_API_KEY')
+    headers = {
+        "accept": "application/json",
+        "content-type": "application/json",
+        "Authorization": f"Bearer {TOGETHER_API_KEY}",
+    }
+
+    try:
+        response = requests.post(url, json=payload, headers=headers)
+        response.raise_for_status()  # Ensure HTTP request was successful
+        response_data = response.json()
+        diagnosis = response_data["choices"][0]["message"]["content"]
+    except requests.exceptions.RequestException as e:
+        diagnosis = "Sorry, I couldn't perform the diagnosis. Please try again later."
+    
+    return diagnosis
 
 # Create sidebar with model selection dropdown and reset button
 selected_model = st.sidebar.selectbox("Select Model", list(model_pre_instructions.keys()))