Update app.py

app.py

@@ -4,45 +4,27 @@ import requests
 # API URL
 API_URL = "https://startrz-proagents.hf.space/api/v1/prediction/7fae8aea-4e25-4feb-8af3-f08e2cd0c020"
 
+# Function to query the API
 def query(payload):
-    """
-    Query the API with the given payload.
-    """
-    try:
-        response = requests.post(API_URL, json=payload)
-        response.raise_for_status()  # Raise error for bad responses (4xx, 5xx)
-        return response.json()
-    except requests.exceptions.HTTPError as http_err:
-        return {"error": f"HTTP error occurred: {http_err}"}
-    except requests.exceptions.RequestException as req_err:
-        return {"error": f"Request error occurred: {req_err}"}
+    response = requests.post(API_URL, json=payload)
+    return response.json()
 
-# Streamlit App
-st.title("Devi Research Task Interface")
-st.markdown("""
-Submit your question or task, and get an intelligent response from the system.
-""")
+# Streamlit app
+def main():
+    st.title("DEVI RESEARCH")
 
-# Input Field
-question = st.text_area("Enter your question or task:", help="Type your query here.")
+    # Input box for user query
+    user_input = st.text_input("Ask a question:")
 
-# Submit Button
-if st.button("Submit"):
-    if question.strip():
-        # Query the API
-        st.markdown(f"**Submitted Question:** {question}")
-        response = query({"question": question})
+    if user_input:
+        # Send the user input to the API
+        response = query({"question": user_input})
 
-        # Display the Result
-        if "error" in response:
-            st.error(f"Error: {response['error']}")
+        # Display the response from the API
+        if response.get("answer"):
+            st.write(f"Response: {response['answer']}")
         else:
-            answer = response.get("answer", "No answer returned.")
-            st.success("Response received:")
-            st.markdown(f"**Answer:** {answer}")
-    else:
-        st.warning("Please enter a question or task before submitting.")
+            st.write("No answer received, please try again.")
 
-# Footer
-st.markdown("---")
-st.markdown("Built with ❤️ using [Streamlit](https://streamlit.io).")
+if __name__ == "__main__":
+    main()