Spaces:
Sleeping
Sleeping
File size: 5,123 Bytes
6c4d96e 99d439f 6c4d96e 99d439f 6c4d96e 99d439f 6c4d96e 99d439f 6c4d96e 99d439f 6c4d96e 99d439f 6c4d96e 99d439f 6c4d96e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 |
import streamlit as st
import time
import random
import json
from datetime import datetime
import pytz
import platform
import uuid
import extra_streamlit_components as stx
import requests
from urllib.parse import quote
# Set page config
st.set_page_config(page_title="Personalized Real-Time Chat with ArXiv Search", page_icon="💬", layout="wide")
# Initialize cookie manager
cookie_manager = stx.CookieManager()
# File to store chat history and user data
CHAT_FILE = "chat_history.txt"
# Function to save chat history and user data to file
def save_data():
with open(CHAT_FILE, 'w') as f:
json.dump({
'messages': st.session_state.messages,
'users': st.session_state.users
}, f)
# Function to load chat history and user data from file
def load_data():
try:
with open(CHAT_FILE, 'r') as f:
data = json.load(f)
st.session_state.messages = data['messages']
st.session_state.users = data['users']
except FileNotFoundError:
st.session_state.messages = []
st.session_state.users = []
# Load data at the start
load_data()
# Function to get or create user
def get_or_create_user():
user_id = cookie_manager.get(cookie='user_id')
if not user_id:
user_id = str(uuid.uuid4())
cookie_manager.set('user_id', user_id)
user = next((u for u in st.session_state.users if u['id'] == user_id), None)
if not user:
user = {
'id': user_id,
'name': random.choice(['Alice', 'Bob', 'Charlie', 'David', 'Eve', 'Frank', 'Grace', 'Henry']),
'browser': f"{platform.system()} - {st.session_state.get('browser_info', 'Unknown')}"
}
st.session_state.users.append(user)
save_data()
return user
# Initialize session state
if 'messages' not in st.session_state:
st.session_state.messages = []
if 'users' not in st.session_state:
st.session_state.users = []
if 'current_user' not in st.session_state:
st.session_state.current_user = get_or_create_user()
# ArXiv search function
def search_arxiv(query):
base_url = "http://export.arxiv.org/api/query?"
search_query = f"search_query=all:{quote(query)}&start=0&max_results=5"
response = requests.get(base_url + search_query)
if response.status_code == 200:
results = []
for entry in response.text.split('<entry>')[1:]:
title = entry.split('<title>')[1].split('</title>')[0]
summary = entry.split('<summary>')[1].split('</summary>')[0]
link = entry.split('<id>')[1].split('</id>')[0]
results.append(f"Title: {title}\nSummary: {summary}\nLink: {link}\n")
return "\n".join(results)
else:
return "Error fetching results from ArXiv."
# Sidebar for user information and settings
with st.sidebar:
st.title("User Info")
st.write(f"Current User: {st.session_state.current_user['name']}")
st.write(f"Browser: {st.session_state.current_user['browser']}")
new_name = st.text_input("Change your name:")
if st.button("Update Name"):
if new_name:
for user in st.session_state.users:
if user['id'] == st.session_state.current_user['id']:
user['name'] = new_name
st.session_state.current_user['name'] = new_name
save_data()
st.success(f"Name updated to {new_name}")
break
st.title("Active Users")
for user in st.session_state.users:
st.write(f"{user['name']} ({user['browser']})")
# Main chat area
st.title("Personalized Real-Time Chat with ArXiv Search")
# Display chat messages
chat_container = st.container()
# Input for new message
new_message = st.text_input("Type your message or ArXiv search query:")
if st.button("Send / Search"):
if new_message:
timestamp = datetime.now(pytz.utc).strftime('%Y-%m-%d %H:%M:%S %Z')
# Check if it's an ArXiv search query
if new_message.lower().startswith("arxiv:"):
query = new_message[6:].strip()
search_results = search_arxiv(query)
st.session_state.messages.append({
'user': 'ArXiv Bot',
'message': f"Search results for '{query}':\n\n{search_results}",
'timestamp': timestamp
})
else:
st.session_state.messages.append({
'user': st.session_state.current_user['name'],
'message': new_message,
'timestamp': timestamp
})
save_data()
st.experimental_rerun()
# Function to display chat messages
def display_messages():
for msg in st.session_state.messages:
with chat_container.container():
st.write(f"**{msg['user']}** ({msg['timestamp']}): {msg['message']}")
# Display messages
display_messages()
# Polling for updates
if st.button("Refresh Chat"):
load_data()
st.experimental_rerun()
# Auto-refresh (note: this will refresh the entire app)
time.sleep(5)
st.experimental_rerun() |