Update app.py

app.py

@@ -2,22 +2,29 @@ import gradio as gr
 import requests
 import os
 
-# Set your Hugging Face API key
+# Fetch Hugging Face API key from environment variables
 API_KEY = os.getenv("HF_API_KEY")
 
-# Hugging Face Model Endpoint (Change to your preferred LLM)
+# Define the Hugging Face model endpoint
 MODEL_NAME = "mistralai/Mistral-7B-Instruct"
+API_URL = f"https://api-inference.huggingface.co/models/{MODEL_NAME}"
 
-# Function to get response from Hugging Face LLM
 def chat_with_llm(prompt):
+    """Sends a prompt to the Hugging Face model and returns the response."""
+    if not API_KEY:
+        return "Error: API key is missing. Please set 'HF_API_KEY' in your environment variables."
+
     headers = {"Authorization": f"Bearer {API_KEY}"}
     payload = {"inputs": prompt}
-    response = requests.post(
-        f"https://api-inference.huggingface.co/models/{MODEL_NAME}",
-        headers=headers,
-        json=payload
-    )
-    return response.json()[0]["generated_text"]
+
+    try:
+        response = requests.post(API_URL, headers=headers, json=payload)
+        response.raise_for_status()  # Raise error for HTTP issues
+        result = response.json()
+        return result[0]["generated_text"] if result else "No response from model."
+    
+    except requests.exceptions.RequestException as e:
+        return f"API Error: {str(e)}"
 
 # Create Gradio Chat UI
 iface = gr.Interface(
@@ -28,5 +35,5 @@ iface = gr.Interface(
     description="A free AI chatbot using Hugging Face's API. Supports multiple LLMs!",
 )
 
-# Launch App
-iface.launch()
+# Launch the app
+iface.launch()