Update app.py

app.py

@@ -226,10 +226,10 @@ with st.sidebar:
         step=1
     )
 
-# API key input
+    # API key input
     api_key = "AIzaSyB93eoXaU48hrJ2hojk-2gQPj28dGVSIbQ"
 
-# Configure Gemini
+    # Configure Gemini
     genai.configure(api_key=api_key)
 
     # Create the model
@@ -258,7 +258,13 @@ if st.button("Generate"):
         try:
             response = model.generate_content(prompt)
             output = response.text.strip()
-                # Display result
+            # 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)}")