Update app.py

app.py

@@ -38,36 +38,85 @@ repo = Repository(
     local_dir="que_gen_logs", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 )
 
+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
+        return result
+    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]
+        gateway = gw[2]
+        host = socket.gethostname()
+        result['ip_addr'] = ipaddr
+        result['host'] = host
+        return result
+    else:
+        result = os.name + " not supported yet."
+        return result
+        
 def generate_questions(article,num_que):
     result = ''
-    print("num_que    :", num_que)
-    if num_que == None or num_que == '':
-        num_que = 5
-    else:
-        num_que = num_que
-    generated_questions_list = qg.generate(article, num_questions=int(num_que))
-    summarized_data = {
-        "generated_questions" : generated_questions_list
-    }
-    generated_questions = summarized_data.get("generated_questions",'')
-    add_csv = [article, generated_questions, num_que]
-    with open(DATA_FILE, "a") as f:
-        writer = csv.writer(f)
-        # write the data
-        writer.writerow(add_csv)
-        commit_url = repo.push_to_hub()
-        print(commit_url)
-    # with open(DATA_FILE, "r") as file:
-    #     data = json.load(file)
-    # data.append(entry)
-    # with open(DATA_FILE, "w") as file:
-    #     json.dump(data, file)
-    #     commit_url = repo.push_to_hub()
-    for q in generated_questions:
-        print(q)
-        result = result + q + '\n'
+    try:
+        if num_que == None or num_que == '':
+            num_que = 5
+        else:
+            num_que = num_que
+        generated_questions_list = qg.generate(article, num_questions=int(num_que))
+        summarized_data = {
+            "generated_questions" : generated_questions_list
+        }
+        generated_questions = summarized_data.get("generated_questions",'')
+            
+        for q in generated_questions:
+            print(q)
+            result = result + q + '\n'
+         save_data_and_sendmail(article,generated_questions,num_que,result)
+        return result
+    except Exception as e:
+        return "Error while generating question -->" + e
+"""
+Save generated details
+"""
+def save_data_and_sendmail(article,generated_questions,num_que,result):
+    try:
+        article = ''
+        generated_questions = []
+        num_que = ''
+        result = ''
+        
+        add_csv = [article, generated_questions, num_que]
+        with open(DATA_FILE, "a") as f:
+            writer = csv.writer(f)
+            # write the data
+            writer.writerow(add_csv)
+            commit_url = repo.push_to_hub()
+            print("commit data   :",commit_url)
+    except Exception as e:
+        return "Error while storing data -->" + e
+        
+    try:  
+        hostname = get_device_ip_address()
+        url = 'http://pragnakalpdev35.pythonanywhere.com/HF_space_que_gen'
+        myobj = {'article': article,'gen_que':result,'ip_addr':hostname.get("ip_addr",""),'host':hostname.get("host","")}
+        x = requests.post(url, json = myobj)
+        print(x)   
+        # with open(DATA_FILE, "r") as file:
+        #     data = json.load(file)
+        # data.append(entry)
+        # with open(DATA_FILE, "w") as file:
+        #     json.dump(data, file)
+        #     commit_url = repo.push_to_hub()
+    except Exception as e:
+        return "Error while sending mail"
         
-    return result
+    return "Successfully save data"
 
 ## design 1
 inputs=gr.Textbox(lines=5, label="Article/Text",elem_id="inp_div")