Update app.py

app.py

@@ -1,5 +1,5 @@
 import streamlit as st
-import requests
+from huggingface_hub import InferenceClient
 from datetime import datetime
 import pandas as pd
 import os
@@ -72,20 +72,27 @@ def get_hf_api_token():
     # For local development, check environment variable
     return os.environ.get("HF_API_TOKEN")
 
-# Function to analyze journal entries
-def analyze_journal(entry, model_name="deepseek-ai/DeepSeek-R1"):
+# Initialize the InferenceClient
+@st.cache_resource
+def get_inference_client():
     api_token = get_hf_api_token()
     if not api_token:
-        return "Error: No API token found. Please configure your Hugging Face API token in the Spaces secrets."
-    
-    API_URL = f"https://api-inference.huggingface.co/models/{model_name}"
+        st.error("No Hugging Face API token found. Please configure it in your Spaces secrets.")
+        return None
     
-    headers = {
-        "Authorization": f"Bearer {api_token}",
-        "Content-Type": "application/json",
-    }
+    return InferenceClient(
+        provider="nebius",
+        api_key=api_token,
+    )
+
+# Function to analyze journal entries
+def analyze_journal(entry, model_name="deepseek-ai/DeepSeek-R1"):
+    client = get_inference_client()
+    if not client:
+        return "Error: Could not initialize the Inference Client."
     
-    prompt = f"""You are a compassionate mental wellness assistant. Analyze the following journal entry with empathy and provide helpful insights.
+    try:
+        prompt = f"""You are a compassionate mental wellness assistant. Analyze the following journal entry with empathy and provide helpful insights.
 
 Journal entry: "{entry}"
 
@@ -96,27 +103,20 @@ Please provide:
 
 Format your response clearly with these exact headings (Emotional tone, Recurring themes, Advice) but without using JSON.
 """
-    
-    payload = {
-        "inputs": prompt,
-        "parameters": {
-            "max_new_tokens": 300,
-            "temperature": 0.7,
-            "top_p": 0.95,
-            "do_sample": True
-        }
-    }
-    
-    try:
-        response = requests.post(API_URL, headers=headers, json=payload)
-        response.raise_for_status()
-        result = response.json()
         
-        # Extract the text from the response
-        if isinstance(result, list) and len(result) > 0:
-            output = result[0].get("generated_text", "")
-        else:
-            output = str(result)
+        completion = client.chat.completions.create(
+            model=model_name,
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt
+                }
+            ],
+            max_tokens=512,
+            temperature=0.7,
+        )
+        
+        output = completion.choices[0].message.content
         
         # Format the output nicely
         formatted_output = ""
@@ -135,13 +135,9 @@ Format your response clearly with these exact headings (Emotional tone, Recurrin
         
         return formatted_output if formatted_output else output
         
-    except requests.exceptions.RequestException as e:
-        st.error(f"API Error: {str(e)}")
-        if hasattr(e, 'response') and e.response is not None:
-            st.error(f"Response: {e.response.text}")
-        return f"Error connecting to the language model: {str(e)}"
     except Exception as e:
-        return f"An unexpected error occurred: {str(e)}"
+        st.error(f"API Error: {str(e)}")
+        return f"Error analyzing journal entry: {str(e)}"
 
 # Function to save an entry
 def save_entry(entry, analysis):
@@ -186,7 +182,7 @@ with st.sidebar:
     st.header("Settings")
     model_choice = st.selectbox(
         "Language Model",
-        ["deepseek-ai/deepseek-v1", "meta-llama/Llama-2-13b-chat-hf", "google/flan-t5-xxl"],
+        ["deepseek-ai/DeepSeek-R1", "meta-llama/Llama-3-8b-chat", "microsoft/phi-3-mini-4k-instruct"],
         index=0,
         help="Select which AI model to use for analysis"
     )
@@ -348,4 +344,7 @@ st.markdown("---")
 st.caption("""
 **Disclaimer**: This app is for educational purposes only and is not a substitute for professional mental health advice, diagnosis, or treatment. 
 Always seek the advice of your physician or other qualified health provider with any questions regarding a medical condition.
-""")
+""")
+     
+        
+