Update app.py

app.py

@@ -130,41 +130,38 @@ def get_location(ip_addr):
     print("response======>>",response)
     return response
     
-def generate_emotion(article):
-    text = "Input sentence: "
-    text += article
-    
-    model_ner = Ner(bert_ner_model)
-    
-    output = model_ner.predict(text)
-    print(output)
-    k = 0
-    Entities_Found.clear()
-    Entity_Types.clear()
-    save_data_and_sendmail(article,output)
-    for i in output:
-        for j in i:
-            if k == 0:
-                Entities_Found.append(j)
-                k += 1
-            else:
-                Entity_Types.append(j)
-                k = 0
-    result = {'Entities Found':Entities_Found, 'Entity Types':Entity_Types}  
-    
-    return pd.DataFrame(result)
+def generate_ner(article):
+    result = {'Entities Found':[], 'Entity Types':[]}
+    if article.strip():
+        text = "Input sentence: "
+        text += article
+        
+        model_ner = Ner(bert_ner_model)
+        
+        output = model_ner.predict(text)
+        print(output)
+        k = 0
+        Entities_Found.clear()
+        Entity_Types.clear()
+        save_data_and_sendmail(article,output)
+        for i in output:
+            for j in i:
+                if k == 0:
+                    Entities_Found.append(j)
+                    k += 1
+                else:
+                    Entity_Types.append(j)
+                    k = 0
+        result = {'Entities Found':Entities_Found, 'Entity Types':Entity_Types}  
+        return pd.DataFrame(result)
+    else:
+        raise gr.Error("Please enter text in inputbox!!!!")
 
 
 def save_data_and_sendmail(article,output):
     try:
         print("welcome")
         ip_address = ''
-        # ip_address = get_device_ip_address()
-        # ip_address=str(request.remote_addr)
-        # print(ip_address)
-        # location = get_location(ip_address)
-        # location = get_location(ip_address)
-        # print(ip_address)
 
         ip_address= getIP()
         print(ip_address)
@@ -194,7 +191,7 @@ def save_data_and_sendmail(article,output):
         return "Error while sending mail" + str(e)
         
 input=gr.Textbox(lines=3, value=input_value, label="Input Text")
-output = [gr.Dataframe(row_count = (2, "dynamic"), col_count=(2, "fixed"), headers=["Entities Found","Entity Types"], lable="Here is the result")]
+output = [gr.Dataframe(row_count = (2, "dynamic"), col_count=(2, "fixed"), headers=["Entities Found","Entity Types"], lable="Here is the result"),wrap=True]
 # with gr.Blocks(css=".gradio-container {background-color: lightgray}") as demo:
 #     gr.Markdown("<h1 style='text-align: center;'>"+ "Named Entity Recognition Using BERT" + "</h1><br/><br/>")
 #     with gr.Row():
@@ -215,14 +212,15 @@ output = [gr.Dataframe(row_count = (2, "dynamic"), col_count=(2, "fixed"), heade
 # demo.launch()
 
 demo = gr.Interface(
-    generate_emotion,
+    generate_ner,
     input,
     output,
     title="Named Entity Recognition Using BERT",
     css=".gradio-container {background-color: lightgray}  #inp_div {background-color: #7FB3D5;",
-    article="""Feel free to give us your [feedback](https://www.pragnakalp.com/contact/) on this NER demo. For all your Named Entity Recognition related 
-            requirements, we are here to help you. Email us your requirement at [letstalk@pragnakalp.com]("mailto:letstalk@pragnakalp.com").
-            And don't forget to check out more interesting [NLP services](https://www.pragnakalp.com/services/natural-language-processing-services/) we are offering.
-                                        <p style='text-align: center;'>Developed by :[ Pragnakalp Techlabs](https://www.pragnakalp.com)</p>"""
+    article="""<p style='text-align: center;'>Feel free to give us your <a href="https://www.pragnakalp.com/contact/" target="_blank">feedback</a> on this NER demo. 
+                For all your Named Entity Recognition related requirements, we are here to help you. Email us your requirement at
+                <a href="mailto:letstalk@pragnakalp.com" target="_blank">letstalk@pragnakalp.com</a>. And don't forget to check out more interesting 
+                <a href="https://www.pragnakalp.com/services/natural-language-processing-services/" target="_blank">NLP services</a> we are offering.
+                    <p style='text-align: center;'>Developed by :<a href="https://www.pragnakalp.com" target="_blank"> Pragnakalp Techlabs</a></p>"""
 )
 demo.launch()