Update app.py

app.py

@@ -4,28 +4,29 @@ import requests
 # API URL
 API_URL = "https://startrz-devi.hf.space/api/v1/prediction/e54adffc-ae77-42e5-9fc0-c4584e081093"
 
-# Function to query the API and extract information
+# Function to query the API and extract required fields
 def query(payload):
     try:
         response = requests.post(API_URL, json=payload)
         response.raise_for_status()  # Raise an error for bad HTTP responses
         data = response.json()  # Parse the JSON response
 
-        # Extract toolOutput if present
+        # Extract toolOutput if present in the response
         tool_outputs = []
         if "agentReasoning" in data:
             for reasoning in data["agentReasoning"]:
                 if "usedTools" in reasoning:
                     tool_outputs.extend(reasoning["usedTools"])
 
-        # Extract title and content from "text"
-        title, content = None, None
+        # Extract title and content from "text" field
+        title = None
+        content = None
         if "text" in data:
             text_data = data["text"]
-            title = text_data.get("title", "No title available")
-            content = text_data.get("content", "No content available")
+            title = text_data.get("title")
+            content = text_data.get("content")
 
-        # Return all extracted data
+        # Return the full response, toolOutput, title, and content
         return {
             "raw_response": data,
             "tool_output": tool_outputs,
@@ -52,7 +53,7 @@ def main():
                     "question": user_input,
                 })
 
-            # Display the extracted toolOutput
+            # Display the extracted toolOutput on the main page
             st.subheader("Extracted Tool Output:")
             tool_output = response.get("tool_output", [])
             if tool_output:
@@ -60,14 +61,20 @@ def main():
             else:
                 st.write("No toolOutput data found in the response.")
 
-            # Display the extracted title and content
+            # Display the extracted title and content on the main page
             st.subheader("Title:")
-            title = response.get("title", "No title available")
-            st.write(title)
+            title = response.get("title")
+            if title:
+                st.write(title)
+            else:
+                st.write("No title found in the response.")
 
             st.subheader("Content:")
-            content = response.get("content", "No content available")
-            st.write(content)
+            content = response.get("content")
+            if content:
+                st.write(content)
+            else:
+                st.write("No content found in the response.")
 
             # Display the raw JSON response in the sidebar
             st.sidebar.header("Raw JSON Response")