Update app.py

app.py

@@ -7,6 +7,9 @@ from transformers import TFAutoModel, AutoTokenizer
 import numpy as np
 import pandas as pd
 import faiss
+import datetime
+import time
+
 
 try:
     nlp = spacy.load("en_core_web_sm")
@@ -110,8 +113,6 @@ def extracted_query_embeddings(queries, max_length=64):
         verbose=1)
     return query_embeddings
 
-#Wikipedia API:
-
 def get_pagetext(page):
     s = str(page).replace("/t","")
     return s
@@ -119,3 +120,97 @@ def get_pagetext(page):
 def get_wiki_summary(search):
     wiki_wiki = wikipediaapi.Wikipedia('en')
     page = wiki_wiki.page(search)                                   
+
+
+def get_wiki_summaryDF(search):
+    wiki_wiki = wikipediaapi.Wikipedia('en')
+    page = wiki_wiki.page(search)
+
+    isExist = page.exists()
+    if not isExist:
+        return isExist, "Not found", "Not found", "Not found", "Not found"
+
+    pageurl = page.fullurl
+    pagetitle = page.title
+    pagesummary = page.summary[0:60]
+    pagetext = get_pagetext(page.text)
+
+    backlinks = page.backlinks
+    linklist = ""
+    for link in backlinks.items():
+      pui = link[0]
+      linklist += pui + " ,  "
+      a=1 
+      
+    categories = page.categories
+    categorylist = ""
+    for category in categories.items():
+      pui = category[0]
+      categorylist += pui + " ,  "
+      a=1     
+    
+    links = page.links
+    linklist2 = ""
+    for link in links.items():
+      pui = link[0]
+      linklist2 += pui + " ,  "
+      a=1 
+      
+    sections = page.sections
+    
+    ex_dic = {
+      'Entity' : ["URL","Title","Summary", "Text", "Backlinks", "Links", "Categories"],
+      'Value': [pageurl, pagetitle, pagesummary, pagetext, linklist,linklist2, categorylist ]
+    }
+
+    df = pd.DataFrame(ex_dic)
+    
+    return df
+
+
+def save_message(name, message):
+    now = datetime.datetime.now()
+    timestamp = now.strftime("%Y-%m-%d %H:%M:%S")
+    with open("chat.txt", "a") as f:
+        f.write(f"{timestamp} - {name}: {message}\n")
+
+def main():
+    st.title("Streamlit Chat")
+
+    name = st.text_input("Name")
+    message = st.text_input("Message")
+    if st.button("Submit"):
+        
+        # wiki
+        df = get_wiki_summaryDF(message)
+        
+        save_message(name, message)
+        save_message(name, df)
+        
+        st.text("Message sent!")
+
+    
+    st.text("Chat history:")
+    with open("chat.txt", "a+") as f:
+        f.seek(0)
+        chat_history = f.read()
+    st.text(chat_history)
+
+    countdown = st.empty()
+    t = 60
+    while t:
+        mins, secs = divmod(t, 60)
+        countdown.text(f"Time remaining: {mins:02d}:{secs:02d}")
+        time.sleep(1)
+        t -= 1
+        if t == 0:
+            countdown.text("Time's up!")
+            with open("chat.txt", "a+") as f:
+                f.seek(0)
+                chat_history = f.read()
+            st.text(chat_history)
+            t = 60
+
+if __name__ == "__main__":
+    main()
+