Update app.py

app.py

@@ -5,6 +5,25 @@ import json
 # API URL
 API_URL = "https://startrz-devi.hf.space/api/v1/prediction/e54adffc-ae77-42e5-9fc0-c4584e081093"
 
+def safe_get(obj, *keys, default=None):
+    """
+    Safely navigate through nested dictionaries
+    
+    Args:
+        obj: Starting object
+        *keys: Sequence of keys to navigate
+        default: Value to return if navigation fails
+    
+    Returns:
+        Value at the specified path or default
+    """
+    try:
+        for key in keys:
+            obj = obj[key]
+        return obj
+    except (TypeError, KeyError, IndexError):
+        return default
+
 def parse_tool_details(tool):
     """
     Parse tool details with robust handling of different input types
@@ -15,17 +34,30 @@ def parse_tool_details(tool):
     Returns:
         dict: Parsed tool details with consistent structure
     """
+    # Ensure tool is a dictionary
+    if not isinstance(tool, dict):
+        return {
+            "tool": "Unknown Tool",
+            "toolInput": "Invalid tool data",
+            "toolOutput": "No output available"
+        }
+
     # Parse toolInput
     input_value = ""
-    if isinstance(tool.get("toolInput"), dict):
-        input_value = tool["toolInput"].get("input", "")
+    tool_input = tool.get("toolInput", {})
+    
+    if isinstance(tool_input, dict):
+        input_value = tool_input.get("input", "")
         # Fallback to full input dict as string if no 'input' key
         if not input_value:
-            input_value = json.dumps(tool["toolInput"], indent=2)
-    elif isinstance(tool.get("toolInput"), str):
-        input_value = tool["toolInput"]
+            try:
+                input_value = json.dumps(tool_input, indent=2)
+            except Exception:
+                input_value = str(tool_input)
+    elif isinstance(tool_input, str):
+        input_value = tool_input
     else:
-        input_value = str(tool.get("toolInput", "No input details"))
+        input_value = str(tool_input) if tool_input is not None else "No input details"
 
     # Parse toolOutput
     output_value = tool.get("toolOutput")
@@ -71,10 +103,14 @@ def query(payload):
         tool_details = []
         
         # Handle different potential response structures
-        if "agentReasoning" in data:
-            for reasoning in data.get("agentReasoning", []):
-                # Extract used tools from each reasoning step
-                for tool in reasoning.get("usedTools", []):
+        agent_reasoning = safe_get(data, "agentReasoning", default=[])
+        
+        for reasoning in agent_reasoning:
+            # Safely extract used tools
+            used_tools = safe_get(reasoning, "usedTools", default=[])
+            
+            for tool in used_tools:
+                if tool is not None:
                     parsed_tool = parse_tool_details(tool)
                     tool_details.append(parsed_tool)
         
@@ -129,7 +165,7 @@ def main():
             if tool_details:
                 # Create tabs for each resource
                 tabs = st.tabs([
-                    f"{idx+1}. {tool['tool']}" 
+                    f"{idx+1}. {tool.get('tool', 'Unknown')}" 
                     for idx, tool in enumerate(tool_details)
                 ])
 
@@ -137,11 +173,11 @@ def main():
                 for idx, (tool, tab) in enumerate(zip(tool_details, tabs)):
                     with tab:
                         st.subheader("Research Input")
-                        st.code(tool['toolInput'], language=None)
+                        st.code(tool.get('toolInput', 'No input'), language=None)
                         
                         st.subheader("Research Findings")
                         # Use st.code for better formatting
-                        st.code(tool['toolOutput'], language=None)
+                        st.code(tool.get('toolOutput', 'No output'), language=None)
 
             else:
                 st.info("No resources found for this query.")