Update app.py

app.py

@@ -7,12 +7,9 @@ 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, 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"}
+        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
     except requests.exceptions.RequestException as e:
         return {"error": str(e)}
 
@@ -26,22 +23,19 @@ def main():
 
     # Add a button to submit the query
     if st.button("Submit"):
-        if user_input.strip():
+        if user_input:
             # Send the user input to the API
             with st.spinner("Fetching response..."):
                 response = query({
-                    "question": user_input.strip(), 
-                    "streaming": True  # Verify if this parameter is supported
+                    "question": user_input, 
+                    "streaming": True
                 })
 
             # Display the raw JSON response
-            if "error" in response:
-                st.error(f"Error: {response['error']}")
-            else:
-                st.subheader("Raw JSON Response:")
-                st.json(response)
+            st.subheader("Raw JSON Response:")
+            st.json(response)
         else:
             st.warning("Please enter a question before submitting.")
 
 if __name__ == "__main__":
-    main()
+    main()