Update disambiguation/disambiguate.py

disambiguation/disambiguate.py

@@ -18,25 +18,47 @@ import regex_spatial
 import streamlit as st
 urllib3.disable_warnings(urllib3.exceptions.InsecureRequestWarning)
 
+BACKEND_URL = "https://SpatialWebAgent-dockerb22.hf.space/api/predict/"
 
+import requests
+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)}"}
 
 
 
 def dismabiguate_entities(doc, ent, ase, level_1, level_2, level_3, midmid):
     return get_coordinates(ent, ase, level_1, level_2, level_3, midmid)
 def get_coordinates(ent, ase, level_1, level_2, level_3, midmid):
-    request_url = 'https://nominatim.openstreetmap.org/search.php?q='+ase+'&polygon_geojson=1&accept-language=en&format=jsonv2'
-    headers = {
-        "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/18.3 Safari/605.1.15"
-    }
-    time.sleep(2)
-    page = requests.get(request_url, headers=headers, verify=False)
-
-    json_content = json.loads(page.content)
-    # json_content = json.loads(aa)
-    print(json_content, 'jjjjj')
-    all_coordinates = json_content[0]['geojson']['coordinates'][0]
-    centroid = (float(json_content[0]['lon']), float(json_content[0]['lat']))
+    # request_url = 'https://nominatim.openstreetmap.org/search.php?q='+ase+'&polygon_geojson=1&accept-language=en&format=jsonv2'
+    # headers = {
+    #     "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/605.1.15 (KHTML, like Gecko) Version/18.3 Safari/605.1.15"
+    # }
+    # time.sleep(2)
+    # page = requests.get(request_url, headers=headers, verify=False)
+
+    # json_content = json.loads(page.content)
+    # print(json_content, 'jjjjj')
+    # all_coordinates = json_content[0]['geojson']['coordinates'][0]
+    # centroid = (float(json_content[0]['lon']), float(json_content[0]['lat']))
+
+    res = call_backend(ase)
+    all_coordinates = res[0]
+    centroid = res[1]
+    
     for p in all_coordinates:
         p2 = (p[0], p[1])
         angle = geoutil.calculate_bearing(centroid, p2)