Upload app.py with huggingface_hub

app.py

@@ -429,14 +429,33 @@ def parse_thinking_response(text: str) -> tuple:
 def parse_tool_use(text: str) -> Optional[Dict[str, Any]]:
     """Parse tool use from model response"""
     try:
-        # Look for JSON tool call pattern
-        json_pattern = r'\{[^{}]*"tool"[^{}]*\}'
-        matches = re.findall(json_pattern, text, re.DOTALL)
-        if matches:
-            for match in matches:
-                parsed = json.loads(match)
-                if "tool" in parsed:
-                    return parsed
+        # Try to find JSON with "tool" key - handle nested braces
+        # First try: the entire text might be JSON
+        text_stripped = text.strip()
+        if text_stripped.startswith("{") and text_stripped.endswith("}"):
+            parsed = json.loads(text_stripped)
+            if "tool" in parsed:
+                return parsed
+
+        # Second try: find JSON object containing "tool"
+        brace_count = 0
+        start_idx = None
+        for i, char in enumerate(text):
+            if char == '{':
+                if brace_count == 0:
+                    start_idx = i
+                brace_count += 1
+            elif char == '}':
+                brace_count -= 1
+                if brace_count == 0 and start_idx is not None:
+                    json_str = text[start_idx:i+1]
+                    try:
+                        parsed = json.loads(json_str)
+                        if "tool" in parsed:
+                            return parsed
+                    except:
+                        pass
+                    start_idx = None
     except:
         pass
     return None