awacke1's picture
Create app.py
8af06a5
raw
history blame
1.72 kB
import streamlit as st
import spacy
import wikipediaapi
import wikipedia
from wikipedia.exceptions import DisambiguationError
from transformers import TFAutoModel, AutoTokenizer
import numpy as np
import pandas as pd
import faiss
import datetime
import time
import random # Import random library
from streamlit.components.v1 import html as html_component # Import html component
# Add a list of random subjects
random_subjects = [
"Computer Science",
"Physics",
"Chemistry",
"Biology",
"History",
"Mathematics",
"Geography",
"Art",
"Music",
"Literature",
]
# ... (rest of the code)
def main():
st.title("Streamlit Chat")
name = st.text_input("Enter your name")
message = st.text_input("Enter a topic to share from Wikipedia")
# If no input is provided, select a random subject
if not message:
message = random.choice(random_subjects)
if st.button("Submit"):
# wiki
df = get_wiki_summaryDF(message)
save_message(name, message)
save_message(name, df.to_string()) # Convert DataFrame to string before saving
st.text("Message sent!")
st.text("Chat history:")
# Display chat history in a data grid
with open("chat.txt", "a+") as f:
f.seek(0)
chat_history = f.read()
# Wrap text in the chat history
wrapped_chat_history = "<br>".join(chat_history.split("\n"))
# Render chat history using the html component
html_component(f"<pre style='white-space: pre-wrap;'>{wrapped_chat_history}</pre>")
countdown = st.empty()
t = 60
while t:
mins, secs = divmod(t, 60)
countdown.text(f"Time remaining: {mins