Update app.py

app.py

@@ -195,7 +195,7 @@ with st.sidebar:
     
     model_name = st.selectbox(
         "Select Model",
-        ["gemini-2.0-pro-exp-02-05", "gemini-exp-1206", "gemini-2.0-flash-lite-preview-02-05", "gemini-1.5-pro", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp-1219"]
+        ["gemini-2.0-pro-exp-02-05", "gemini-exp-1206", "gemini-2.0-flash-lite-preview-02-05", "gemini-1.5-pro", "gemini-2.0-flash-exp", "gemini-2.0-flash-thinking-exp-1219", "learnlm-1.5-pro-experimental"]
     )
     
     temperature = st.slider(
@@ -224,6 +224,7 @@ with st.sidebar:
 
     api_key = "AIzaSyB93eoXaU48hrJ2hojk-2gQPj28dGVSIbQ"
     backup_api_key = "AIzaSyDNWizNQAA0UaflR5f_zTF6FcMSIA5D8RU"
+    backup_api_key_2 = "AIzaSyCrhuU-XT142KWB_zmA807qeJcE0dEjeJw"
 
     genai.configure(api_key=api_key)
 
@@ -254,17 +255,29 @@ if st.button("Generate"):
         except Exception as e:
             if "429" in str(e):
                 try:
-                    # Switch to backup API key
+                    # Switch to first backup API key
                     genai.configure(api_key=backup_api_key)
                     model = genai.GenerativeModel(
                         model_name=model_name,
                         generation_config=generation_config,
                     )
-                    # Retry with new API key
+                    # Retry with first backup API key
                     response = model.generate_content(prompt)
                     output = response.text.strip()
                     st.success(f"Output: {output}")
                 except Exception as backup_error:
-                    st.error(f"Error with backup API key: {str(backup_error)}")
+                    try:
+                        # Switch to second backup API key
+                        genai.configure(api_key=backup_api_key_2)
+                        model = genai.GenerativeModel(
+                            model_name=model_name,
+                            generation_config=generation_config,
+                        )
+                        # Retry with second backup API key
+                        response = model.generate_content(prompt)
+                        output = response.text.strip()
+                        st.success(f"Output: {output}")
+                    except Exception as backup_error_2:
+                        st.error(f"Error with both backup API keys: {str(backup_error_2)}")
             else:
                 st.error(f"An error occurred: {str(e)}")