Update app.py

app.py

@@ -53,8 +53,13 @@ def query(payload, api_url):
     headers = {"Authorization": f"Bearer {st.secrets['HF_TOKEN']}"}
     logger.info(f"Sending request to {api_url} with payload: {payload}")
     response = requests.post(api_url, headers=headers, json=payload)
-    logger.info(f"Received response: {response.status_code}, {response.text}")
-    return response.json()
+    try:
+        response.raise_for_status()
+        logger.info(f"Received response: {response.status_code}, {response.text}")
+        return response.json()
+    except requests.exceptions.RequestException as e:
+        logger.error(f"HTTP Request failed: {e}")
+        return None
 
 # Chat interface
 st.title("🤖 DeepSeek Chatbot")
@@ -101,7 +106,7 @@ if prompt := st.chat_input("Type your message..."):
             output = query(payload, api_url)
 
             # Handle API response
-            if isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
+            if output and isinstance(output, list) and len(output) > 0 and 'generated_text' in output[0]:
                 assistant_response = output[0]['generated_text']
                 logger.info(f"Generated response: {assistant_response}")
 
@@ -110,9 +115,9 @@ if prompt := st.chat_input("Type your message..."):
 
                 st.session_state.messages.append({"role": "assistant", "content": assistant_response})
             else:
-                logger.error(f"Unexpected API response: {output}")
+                logger.error(f"Unexpected API response or error: {output}")
                 st.error("Error: Unable to generate a response. Please try again.")
 
     except Exception as e:
         logger.error(f"Application Error: {str(e)}", exc_info=True)
-        st.error(f"Application Error: {str(e)}")
+        st.error(f"Application Error: {str(e)}")