Update app.py

app.py

@@ -7,9 +7,12 @@ API_URL = "https://startrz-devi.hf.space/api/v1/prediction/e54adffc-ae77-42e5-9f
 # Function to query the API
 def query(payload):
     try:
-        response = requests.post(API_URL, json=payload)
-        response.raise_for_status()  # Raise an error for bad HTTP responses
-        return response.json()  # Return the JSON response
+        response = requests.post(API_URL, json=payload, timeout=30)  # Increased timeout
+        response.raise_for_status()  # Raise for bad HTTP responses
+        try:
+            return response.json()  # Return JSON response
+        except ValueError:
+            return {"error": "Invalid JSON response from API"}
     except requests.exceptions.RequestException as e:
         return {"error": str(e)}
 
@@ -23,19 +26,22 @@ def main():
 
     # Add a button to submit the query
     if st.button("Submit"):
-        if user_input:
+        if user_input.strip():
             # Send the user input to the API
             with st.spinner("Fetching response..."):
                 response = query({
-                    "question": user_input, 
-                    "streaming": True
+                    "question": user_input.strip(), 
+                    "streaming": True  # Verify if this parameter is supported
                 })
 
             # Display the raw JSON response
-            st.subheader("Raw JSON Response:")
-            st.json(response)
+            if "error" in response:
+                st.error(f"Error: {response['error']}")
+            else:
+                st.subheader("Raw JSON Response:")
+                st.json(response)
         else:
             st.warning("Please enter a question before submitting.")
 
 if __name__ == "__main__":
-    main()
+    main()