Update app.py

app.py

@@ -21,7 +21,7 @@ import huggingface_hub
 import socket
 
 HF_TOKEN = os.environ.get("HF_TOKEN")
-DATASET_NAME = "bert_base_ner"
+DATASET_NAME = "bert_based_ner_dataset "
 DATASET_REPO_URL = f"https://huggingface.co/datasets/pragnakalp/{DATASET_NAME}"
 DATA_FILENAME = "bert_base_ner_logs.csv"
 DATA_FILE = os.path.join("bert_base_ner_logs", DATA_FILENAME)
@@ -49,32 +49,45 @@ Entities_Found =[]
 Entity_Types = []
 k = 0
 def get_device_ip_address():
-    result = {}
+    
     if os.name == "nt":
         result = "Running on Windows"
         hostname = socket.gethostname()
         ip_address = socket.gethostbyname(hostname)
-        result['ip_addr'] = ip_address
-        result['host'] = hostname
         print(result)
-        return result
+        return ip_address
     elif os.name == "posix":
         gw = os.popen("ip -4 route show default").read().split()
         s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
         s.connect((gw[2], 0))
-        ipaddr = s.getsockname()[0]
+        ip_address = s.getsockname()[0]
         gateway = gw[2]
         host = socket.gethostname()
-        result['ip_addr'] = ipaddr
-        result['host'] = host
         print(result)
-        return result
+        return ip_address
     else:
         result['id'] = os.name + " not supported yet."
         print(result)
         return result
 
+def get_location(ip_addr):
+    ip=ip_addr
+    # ip=str(request.remote_addr)
+    req_data={
+        "ip":ip,
+        "token":"pkml123"
+    }
+    url = "https://demos.pragnakalp.com/get-ip-location"
 
+    # req_data=json.dumps(req_data)
+    # print("req_data",req_data)
+    headers = {'Content-Type': 'application/json'}
+
+    response = requests.request("POST", url, headers=headers, data=json.dumps(req_data))
+    response = response.json()
+    print("response======>>",response)
+    return response
+    
 def generate_emotion(article):
     text = "Input sentence: "
     text += article
@@ -102,22 +115,20 @@ def generate_emotion(article):
 
 def save_data_and_sendmail(article,output):
     try:
-        print(">>"*30)
-        hostname = {}
-        hostname = get_device_ip_address()
+        ip_address = ''
+        ip_address = get_device_ip_address()
+        location = get_location(ip_address)
         # url = 'https://pragnakalpdev35.pythonanywhere.com/HF_space_que_gen'
         # # url = 'http://pragnakalpdev33.pythonanywhere.com/HF_space_question_generator'
         # myobj = {'article': article,'total_que': num_que,'gen_que':result,'ip_addr':hostname.get("ip_addr",""),'host':hostname.get("host","")}
         # x = requests.post(url, json = myobj) 
+        
         inputdata = article
-        ip = hostname.get("ip_addr","")
-        print(ip)
         url = 'https://pragnakalpdev33.pythonanywhere.com/HF_space_bert_base_ner'
-        # url = 'http://pragnakalpdev33.pythonanywhere.com/HF_space_question_generator'
-        myobj = {'article': article,'gen_text':output,'ip_addr':hostname.get("ip_addr",""),'host':hostname.get("host","")}
+        myobj = {'article': article,'gen_text':output,'ip_addr':ip_address,'loc':location}
         x = requests.post(url, json = myobj) 
         
-        add_csv = [inputdata,output,ip]
+        add_csv = [inputdata,output,ip_address,location]
         with open(DATA_FILE, "a") as f:
             writer = csv.writer(f)
             # write the data