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")
+API_TOKEN = 'hf_'
 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","")
@@ -236,7 +236,7 @@ def extract_spatial_entities(text):
     # st.markdown("123123")
 
     # show_sentence_selector_table(doc_copy)
-
+    pass
 def show_sentence_selector_table(doc_copy):
     st.markdown("**______________________________________________________________________________________**")
     st.markdown("**Sentence Selector for Geographic Composition**")