Update app.py

app.py

@@ -7,9 +7,11 @@ 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", page_icon="💬", layout="wide")
+st.set_page_config(page_title="Personalized Real-Time Chat with ArXiv Search", page_icon="💬", layout="wide")
 
 # Initialize cookie manager
 cookie_manager = stx.CookieManager()
@@ -66,6 +68,23 @@ if 'users' not in st.session_state:
 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")
@@ -82,27 +101,39 @@ with st.sidebar:
                     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")
+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:")
-if st.button("Send"):
+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')
-        st.session_state.messages.append({
-            'user': st.session_state.current_user['name'],
-            'message': new_message,
-            'timestamp': timestamp
-        })
+        
+        # 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()