Update app.py

app.py

@@ -7,7 +7,7 @@ import pymongo
 import torch
 import spacy
 from spacy import displacy
-from pymongo import MongoClient
+# from pymongo import MongoClient
 
 from transformers import AutoTokenizer
 tokenizer = AutoTokenizer.from_pretrained("law-ai/InLegalBERT")
@@ -248,30 +248,30 @@ def judgtext_analysis(text):
   doc.ents = ents
 
   #logic for repository
-  cluster = MongoClient("mongodb+srv://testuser:test123@ner-gradio.mgng1wv.mongodb.net/?retryWrites=true&w=majority")
-  db = cluster["nerdb"]
-  collection = db["named_entities"] 
-  content = displacy.render(doc, style='ent')
-  extsoup = bs(content, 'html.parser')
-
-  txtlist=[]
-  entlist=[]
-  for h in extsoup.findAll('div'):
-    mark = h.findAll('mark')
-    span = h.findAll('span')
-    for i in mark:
-      txt = i.find(text=True)
-      txt = txt.replace("\n","");
-      txt = txt.strip();
-      txtlist.append(str(txt))
-    for i in span:
-      ent = i.find(text=True)
-      entlist.append(str(ent))
-  zipped = zip(txtlist, entlist)
-  ziplist = list(zipped)
-
-  post = {"api_docid":docid, "document_text":finaltext, "named_entities":ziplist, "entities_indices": finalentities}
-  collection.insert_one(post)
+  # cluster = MongoClient("mongodb+srv://testuser:test123@ner-gradio.mgng1wv.mongodb.net/?retryWrites=true&w=majority")
+  # db = cluster["nerdb"]
+  # collection = db["named_entities"] 
+    content = displacy.render(doc, style='ent')
+  # extsoup = bs(content, 'html.parser')
+
+  # txtlist=[]
+  # entlist=[]
+  # for h in extsoup.findAll('div'):
+  #   mark = h.findAll('mark')
+  #   span = h.findAll('span')
+  #   for i in mark:
+  #     txt = i.find(text=True)
+  #     txt = txt.replace("\n","");
+  #     txt = txt.strip();
+  #     txtlist.append(str(txt))
+  #   for i in span:
+  #     ent = i.find(text=True)
+  #     entlist.append(str(ent))
+  # zipped = zip(txtlist, entlist)
+  # ziplist = list(zipped)
+
+  # post = {"api_docid":docid, "document_text":finaltext, "named_entities":ziplist, "entities_indices": finalentities}
+  # collection.insert_one(post)
 
   html = displacy.render(doc, style="ent", page = True)
   html = (