Update app.py

app.py

@@ -9,9 +9,10 @@ import uuid
 import extra_streamlit_components as stx
 import requests
 from urllib.parse import quote
+from openai import OpenAI
 
 # Set page config
-st.set_page_config(page_title="Personalized Real-Time Chat with ArXiv Search", page_icon="💬", layout="wide")
+st.set_page_config(page_title="Personalized Real-Time Chat with ArXiv Search and AI", page_icon="💬", layout="wide")
 
 # Initialize cookie manager
 cookie_manager = stx.CookieManager()
@@ -87,6 +88,27 @@ def search_arxiv(query):
     else:
         return "Error fetching results from ArXiv."
 
+# Initialize OpenAI client
+client = OpenAI(api_key=st.secrets['OPENAI_API_KEY'])
+MODEL = "gpt-4-0125-preview"  # Use the appropriate model
+
+# Function to get AI response
+def get_ai_response(prompt, context=""):
+    try:
+        messages = [
+            {"role": "system", "content": "You are a helpful assistant in a chat room that can also search ArXiv."},
+            {"role": "user", "content": f"Context: {context}\n\nUser Query: {prompt}"}
+        ]
+        response = client.chat.completions.create(
+            model=MODEL,
+            messages=messages,
+            max_tokens=150,
+            temperature=0.7
+        )
+        return response.choices[0].message.content
+    except Exception as e:
+        return f"An error occurred: {str(e)}"
+
 # Sidebar for user information and settings
 with st.sidebar:
     st.title("User Info")
@@ -109,7 +131,7 @@ with st.sidebar:
         st.write(f"{user['name']} ({user['browser']})")
 
 # Main chat area
-st.title("Personalized Real-Time Chat with ArXiv Search")
+st.title("Personalized Real-Time Chat with ArXiv Search and AI")
 
 # Display chat messages
 for message in st.session_state.messages:
@@ -129,13 +151,19 @@ if prompt := st.chat_input("Type your message or ArXiv search query:"):
             with st.spinner("Searching ArXiv..."):
                 search_results = search_arxiv(query)
                 st.markdown(f"Search results for '{query}':\n\n{search_results}")
-        st.session_state.messages.append({"role": "assistant", "content": f"Search results for '{query}':\n\n{search_results}"})
+                
+                # Get AI commentary on the search results
+                ai_commentary = get_ai_response(f"Provide a brief analysis of these ArXiv search results: {search_results}")
+                st.markdown(f"\nAI Analysis:\n{ai_commentary}")
+        
+        st.session_state.messages.append({"role": "assistant", "content": f"Search results for '{query}':\n\n{search_results}\n\nAI Analysis:\n{ai_commentary}"})
     else:
-        # Here you would typically send the message to your chat model and get a response
-        # For this example, we'll just echo the message
+        # Get AI response for regular chat
         with st.chat_message("assistant"):
-            st.markdown(f"You said: {prompt}")
-        st.session_state.messages.append({"role": "assistant", "content": f"You said: {prompt}"})
+            with st.spinner("AI is thinking..."):
+                ai_response = get_ai_response(prompt)
+                st.markdown(ai_response)
+        st.session_state.messages.append({"role": "assistant", "content": ai_response})
 
     save_data()
     st.rerun()