Update app.py

app.py

@@ -3,15 +3,29 @@ import requests
 import os
 
 HF_API_KEY = os.getenv('TOKEN')
-MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B"
+MODEL_NAME = "deepseek-ai/DeepSeek-R1-Distill-Llama-8B" 
 
 def chat(prompt):
+    # Make the API request
     response = requests.post(
         f"https://api-inference.huggingface.co/models/{MODEL_NAME}",
         headers={"Authorization": f"Bearer {HF_API_KEY}"},
         json={"inputs": prompt}
     )
-    return response.json()[0]["generated_text"]
+    
+    # Debug: print the raw response to see the actual structure
+    print(response.json())  # This will help you inspect the returned data
 
-iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="DeepSeek API Chatbot")
+    # Safe check: return the response if it's structured as expected
+    try:
+        # Try accessing the 'generated_text' field, assuming a list-like response
+        generated_text = response.json()[0]["generated_text"]
+        return generated_text
+    except (KeyError, IndexError) as e:
+        # Handle the case where the structure is different
+        print(f"Error: {e}")
+        return f"Error processing response: {response.json()}"
+
+# Set up Gradio interface
+iface = gr.Interface(fn=chat, inputs="text", outputs="text", title="DeepSeek 7B Chatbot (API)")
 iface.launch()