Update app.py

app.py

@@ -562,7 +562,7 @@ def main():
                 elif model_choice == "Claude-3":
                     claude_response = process_with_claude(user_input)
                 else:  # Both
-                    col1, col2 = st.columns(2)
+                    col1, col2, col3 = st.columns(3)
                     with col2:
                         st.subheader("Claude-3.5 Sonnet:")
                         try:
@@ -575,6 +575,12 @@ def main():
                             gpt_response = process_with_gpt(user_input)  
                         except:
                             st.write('GPT 4o out of tokens')
+                    with col3:
+                        st.subheader("Arxiv and Mistral Research:")
+                        with st.spinner("Searching ArXiv..."):
+                            results = search_arxiv(query)
+                            st.markdown(results)
+                            
         # Display Chat History
         st.subheader("Chat History 📜")
         tab1, tab2 = st.tabs(["Claude History", "GPT-4o History"])