Update app.py

app.py

@@ -188,19 +188,16 @@ history = [
   "output: 1. directing his knowingly armed supporters to the capitol",
     ]
 
-# Create the Streamlit interface
 st.title("Pattern Completion")
 
 with st.sidebar:
     st.header("Model Configuration")
     
-    # Model selection dropdown
     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"]
     )
     
-    # Temperature slider
     temperature = st.slider(
         "Temperature",
         min_value=0.0,
@@ -209,7 +206,6 @@ with st.sidebar:
         step=0.1
     )
     
-    # Top_p slider
     top_p = st.slider(
         "Top P",
         min_value=0.0,
@@ -226,13 +222,11 @@ with st.sidebar:
         step=1
     )
 
-    # API key input
     api_key = "AIzaSyB93eoXaU48hrJ2hojk-2gQPj28dGVSIbQ"
+    backup_api_key = "AIzaSyDNWizNQAA0UaflR5f_zTF6FcMSIA5D8RU"
 
-    # Configure Gemini
     genai.configure(api_key=api_key)
 
-    # Create the model
     generation_config = {
         "temperature": temperature,
         "top_p": top_p,
@@ -246,25 +240,31 @@ with st.sidebar:
         generation_config=generation_config,
     )
 
-    # Input text box (now using text_area instead of text_input)
 user_input = st.text_area("Enter your input:", height=150)
 if st.button("Generate"):
     if user_input:
-        # Create prompt with history and new input
         prompt = history.copy()
         prompt.extend([f"input: {user_input}", "output: "])
         print(prompt)
-        # Generate response
+        
         try:
             response = model.generate_content(prompt)
             output = response.text.strip()
-            # Display result
             st.success(f"Output: {output}")
         except Exception as e:
-            api_key = "AIzaSyDNWizNQAA0UaflR5f_zTF6FcMSIA5D8RU"
-            genai.configure(api_key=api_key)
-            model = genai.GenerativeModel(
-                model_name=model_name,
-                generation_config=generation_config,
-            )
-            st.error(f"An error occurred: {str(e)}")
+            if "429" in str(e):
+                try:
+                    # Switch to 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
+                    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)}")
+            else:
+                st.error(f"An error occurred: {str(e)}")