Update 1_SpatialParse.py

1_SpatialParse.py

@@ -11,27 +11,27 @@ from spacy.tokens import Span, Doc, Token
 from utils import geoutil
 import urllib.parse
 import os
-import requests
+# import requests
 
 API_TOKEN = os.getenv("HF_API_TOKEN")
 BACKEND_URL = "https://dsbb0707-dockerb2.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:
-            result = response.json()["data"][0]
-            return f"✅ {result['result']}\n⏰ {result['timestamp']}"
-        return f"❌ Backend Error (HTTP {response.status_code})"
-    except Exception as e:
-        return f"⚠️ Connection Error: {str(e)}"
+# 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:
+#             result = response.json()["data"][0]
+#             return f"✅ {result['result']}\n⏰ {result['timestamp']}"
+#         return f"❌ Backend Error (HTTP {response.status_code})"
+#     except Exception as e:
+#         return f"⚠️ Connection Error: {str(e)}"
 
 
 colors = {'GPE': "#43c6fc", "LOC": "#fd9720", "RSE":"#a6e22d"}
@@ -224,9 +224,9 @@ def extract_spatial_entities(text):
     # # doc.to_disk("saved_doc.spacy")
     # doc.to_disk("/tmp/saved_doc.spacy")
 
-    api_result = call_backend(text)
-    st.markdown(api_result)
-    st.text_area(api_result)
+    # api_result = call_backend(text)
+    # st.markdown(api_result)
+    # st.text_area(api_result)
 
     # html = displacy.render(doc,style="ent", options = options)
     # html = html.replace("\n","")