update fix button issue

app.py

@@ -14,7 +14,7 @@ from huggingface_hub import InferenceClient
 # ============================================
 
 CLASSIFIER_MODEL = "YureiYuri/empathy"
-LLM_MODEL = "openai/gpt-oss-20b"
+LLM_MODEL = "meta-llama/Llama-3.2-3B-Instruct"  # Free model, better than TinyLlama
 
 # ============================================
 # LOAD CLASSIFIER
@@ -146,25 +146,42 @@ def respond(
         yield crisis_msg
         return
     
-    # Generate streaming response
-    client = InferenceClient(token=hf_token.token, model=LLM_MODEL)
-    
-    system_message = get_system_prompt()
-    messages = [{"role": "system", "content": system_message}]
-    messages.extend(history)
-    messages.append({"role": "user", "content": message})
+    # Check if user is authenticated
+    if not hf_token or not hf_token.token:
+        yield "⚠️ Please sign in with your Hugging Face account (click the button in the sidebar) to start chatting."
+        return
     
-    response = ""
-    for msg in client.chat_completion(
-        messages,
-        max_tokens=max_tokens,
-        stream=True,
-        temperature=temperature,
-        top_p=top_p,
-    ):
-        if msg.choices and msg.choices[0].delta.content:
-            response += msg.choices[0].delta.content
-            yield response
+    # Generate streaming response with error handling
+    try:
+        client = InferenceClient(token=hf_token.token, model=LLM_MODEL)
+        
+        system_message = get_system_prompt()
+        messages = [{"role": "system", "content": system_message}]
+        messages.extend(history)
+        messages.append({"role": "user", "content": message})
+        
+        response = ""
+        for msg in client.chat_completion(
+            messages,
+            max_tokens=max_tokens,
+            stream=True,
+            temperature=temperature,
+            top_p=top_p,
+        ):
+            if msg.choices and msg.choices[0].delta.content:
+                response += msg.choices[0].delta.content
+                yield response
+                
+    except Exception as e:
+        error_msg = str(e)
+        print(f"❌ LLM Error: {error_msg}")
+        
+        if "401" in error_msg or "Unauthorized" in error_msg:
+            yield "⚠️ Authentication error. Please sign in with your Hugging Face account using the button in the sidebar."
+        elif "quota" in error_msg.lower() or "rate" in error_msg.lower():
+            yield "⚠️ API rate limit reached. Please try again in a moment."
+        else:
+            yield "I'm here to listen and support you. Could you tell me more about what you're experiencing?"
 
 
 # ============================================
@@ -190,7 +207,11 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
         
         with gr.Column(scale=1):
-            gr.LoginButton()
+            gr.Markdown("### 🔐 Authentication")
+            gr.LoginButton(size="lg")
+            gr.Markdown("*Sign in with Hugging Face to chat*")
+            
+            gr.Markdown("---")
             
             gr.Markdown("### 📊 Emotional Tracking")
             gr.Markdown(
@@ -202,14 +223,14 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             )
             
             metrics_display = gr.Markdown(f"""
-                **Current Mode:** 🟢 SUPPORTIVE
-                
-                **Severity:** 0.0/100
-                - Dejection: 0.0
-                - Mood: 0.0  
-                - Calmness: 0.0
-                
-                *Metrics update as you chat*
+            **Current Mode:** 🟢 SUPPORTIVE
+            
+            **Severity:** 0.0/100
+            - Dejection: 0.0
+            - Mood: 0.0  
+            - Calmness: 0.0
+            
+            *Metrics update as you chat*
             """)
     
     gr.Markdown(