Update app.py

app.py

@@ -563,13 +563,18 @@ def main():
                     claude_response = process_with_claude(user_input)
                 else:  # Both
                     col1, col2 = st.columns(2)
-                    with col1:
-                        st.subheader("GPT-4o Omni:")
-                        gpt_response = process_with_gpt(user_input)
                     with col2:
                         st.subheader("Claude-3.5 Sonnet:")
-                        claude_response = process_with_claude(user_input)
-        
+                        try:
+                            claude_response = process_with_claude(user_input)
+                        except:
+                            st.write('Claude 3.5 Sonnet out of tokens.')
+                    with col1:
+                        st.subheader("GPT-4o Omni:")
+                        try:
+                            gpt_response = process_with_gpt(user_input)  
+                        except:
+                            st.write('GPT 4o out of tokens')
         # Display Chat History
         st.subheader("Chat History 📜")
         tab1, tab2 = st.tabs(["Claude History", "GPT-4o History"])