Update app.py

app.py

@@ -58,7 +58,6 @@ def main():
 
         # Define model priority
         models = ["o1-preview", "o1-preview-2024-09-12","o1-mini","gpt-4o-mini","gpt-3.5-turbo"]  # Add more models as needed
-
         # Get API key
         api_key = get_api_key()
         if not api_key:
@@ -81,6 +80,7 @@ def main():
             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})