Update app.py

app.py

@@ -11,29 +11,22 @@ def query(payload):
         response.raise_for_status()  # Raise an error for bad HTTP responses
         data = response.json()  # Parse the JSON response
 
-        # Extract toolOutput and Resource Name from the response
-        resources = []
+        # Extract toolOutput if present in the response
+        tool_outputs = []
         if "agentReasoning" in data:
             for reasoning in data["agentReasoning"]:
                 if "usedTools" in reasoning:
-                    for tool in reasoning["usedTools"]:
-                        tool_input = tool.get("toolInput", {}).get("input", "")
-                        tool_output = tool.get("toolOutput", "")
-                        if tool_input and tool_output:
-                            resources.append({
-                                "resource_name": tool_input,  # Extracted from toolInput -> input
-                                "resource": tool_output  # Extracted from toolOutput
-                            })
+                    tool_outputs.extend(reasoning["usedTools"])
 
-        # Return the full response and resources
-        return {"raw_response": data, "resources": resources}
+        # Return the full response and toolOutput
+        return {"raw_response": data, "tool_output": tool_outputs}
     except requests.exceptions.RequestException as e:
         return {"error": str(e)}
 
 # Streamlit app
 def main():
-    st.title("Online Resources")
-    st.write("Ask any question, and explore online resources provided based on the response!")
+    st.title("DEVI RESEARCH")
+    st.write("Ask any question, and the API will provide an answer!")
 
     # Input box for user query
     user_input = st.text_input("Enter your question below:")
@@ -47,15 +40,13 @@ def main():
                     "question": user_input,
                 })
 
-            # Display the online resources
-            st.subheader("Online Resources:")
-            resources = response.get("resources", [])
-            if resources:
-                for resource in resources:
-                    with st.expander(resource["resource_name"]):
-                        st.write(resource["resource"])
+            # Display the extracted toolOutput on the main page
+            st.subheader("Extracted Tool Output:")
+            tool_output = response.get("tool_output", [])
+            if tool_output:
+                st.write(tool_output)
             else:
-                st.write("No resources found in the response.")
+                st.write("No toolOutput data found in the response.")
 
             # Display the raw JSON response in the sidebar
             st.sidebar.header("Raw JSON Response")