Update app.py

app.py

@@ -22,16 +22,20 @@ def generate_response(messages, model_name, api_key): # Modified to accept 'mess
         )
         return stream
     except openai.APIError as e:
-        st.error(f"OpenAI API Error with {model_name}: {e}")
+        # Log the error for debugging, but don't display it in the UI
+        print(f"OpenAI API Error with {model_name}: {e}")
         return None
     except openai.RateLimitError as e:
-        st.error(f"OpenAI Rate Limit Error with {model_name}: {e}")
+        # Log the error for debugging, but don't display it in the UI
+        print(f"OpenAI Rate Limit Error with {model_name}: {e}")
         return None
     except openai.AuthenticationError as e:
-        st.error(f"OpenAI Authentication Error with {model_name}: {e}")
+        # Log the error for debugging, but don't display it in the UI
+        print(f"OpenAI Authentication Error with {model_name}: {e}")
         return None
     except Exception as e:
-        st.error(f"An unexpected error occurred with {model_name}: {e}")
+        # Log the error for debugging, but don't display it in the UI
+        print(f"An unexpected error occurred with {model_name}: {e}")
         return None
 
 # Main Streamlit app
@@ -76,11 +80,11 @@ def main():
                             full_response += chunk.choices[0].delta.content
                             message_placeholder.markdown(full_response + "▌")
                     message_placeholder.markdown(full_response)
+                print(f"Using {model} for generation")
                 break # Break after successful response
             full_response = "" # Reset for the next model attempt
 
         if full_response:
-            print(f"Using {model} for generation")
             # Add bot message to state
             st.session_state.messages.append({"role": "assistant", "content": full_response})