Update app.py

app.py

@@ -11,15 +11,20 @@ def query(payload):
         response.raise_for_status()  # Raise an error for bad HTTP responses
         data = response.json()  # Parse the JSON response
 
-        # Extract toolOutput if present in the response
-        tool_outputs = []
+        # Extract toolOutput and tool details
+        tool_details = []
         if "agentReasoning" in data:
             for reasoning in data["agentReasoning"]:
                 if "usedTools" in reasoning:
-                    tool_outputs.extend(reasoning["usedTools"])
+                    for tool in reasoning["usedTools"]:
+                        tool_details.append({
+                            "tool": tool.get("tool", ""),
+                            "toolInput": tool.get("toolInput", {}).get("input", ""),
+                            "toolOutput": tool.get("toolOutput", "")
+                        })
 
-        # Return the full response and toolOutput
-        return {"raw_response": data, "tool_output": tool_outputs}
+        # Return the full response and tool details
+        return {"raw_response": data, "tool_details": tool_details}
     except requests.exceptions.RequestException as e:
         return {"error": str(e)}
 
@@ -40,13 +45,36 @@ def main():
                     "question": user_input,
                 })
 
-            # 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)
+            # Check for errors
+            if "error" in response:
+                st.error(f"Error: {response['error']}")
+                return
+
+            # Display Online Resources section
+            st.subheader("Online Resources")
+            tool_details = response.get("tool_details", [])
+            
+            if tool_details:
+                # Create an expandable section for each resource
+                for idx, tool in enumerate(tool_details, 1):
+                    # Use toolInput as the resource name
+                    resource_name = tool.get("toolInput", f"Resource {idx}")
+                    
+                    # Create an expandable section
+                    with st.expander(resource_name):
+                        # Display the tool output
+                        tool_output = tool.get("toolOutput", "No output available")
+                        
+                        # Check if toolOutput is a list or dict and pretty print
+                        if isinstance(tool_output, (list, dict)):
+                            st.json(tool_output)
+                        else:
+                            st.write(tool_output)
+                        
+                        # Optional: Add tool type information
+                        st.caption(f"Source: {tool.get('tool', 'Unknown')}")
             else:
-                st.write("No toolOutput data found in the response.")
+                st.write("No resources found.")
 
             # Display the raw JSON response in the sidebar
             st.sidebar.header("Raw JSON Response")