Update disambiguation/disambiguate.py

disambiguation/disambiguate.py

@@ -18,24 +18,24 @@ import regex_spatial
 import streamlit as st
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
-# BACKEND_URL = "https://SpatialWebAgent-dockerb22.hf.space/api/predict/"
-
-# def call_backend(input_text):
-#     try:
-#         headers = {
-#             "Authorization": f"Bearer {API_TOKEN}"
-#         }
-#         response = requests.post(
-#             BACKEND_URL,
-#             headers=headers,
-#             json={"data": [input_text]},
-#             timeout=10
-#         )
-#         if response.status_code == 200:
-#             return response.json()  # ✅ 保留原始 JSON 对象 (dict)
-#         return {"error": f"❌ Backend Error (HTTP {response.status_code})"}
-#     except Exception as e:
-#         return {"error": f"⚠️ Connection Error: {str(e)}"}
+BACKEND_URL = "https://SpatialWebAgent-dockerb22.hf.space/api/predict/"
+
+def call_backend(input_text):
+    try:
+        headers = {
+            "Authorization": f"Bearer {API_TOKEN}"
+        }
+        response = requests.post(
+            BACKEND_URL,
+            headers=headers,
+            json={"data": [input_text]},
+            timeout=10
+        )
+        if response.status_code == 200:
+            return response.json()  # ✅ 保留原始 JSON 对象 (dict)
+        return {"error": f"❌ Backend Error (HTTP {response.status_code})"}
+    except Exception as e:
+        return {"error": f"⚠️ Connection Error: {str(e)}"}