Update app.py

app.py

@@ -42,7 +42,10 @@ def generate_code(prompt):
     
     try:
         genai.configure(api_key=GEMINI_API_KEY)
-        model = genai.GenerativeModel('gemini-1.5-flash')
+        # Get available models and use the first flash model
+        models = [m.name for m in genai.list_models() if 'generateContent' in m.supported_generation_methods]
+        model_name = next((m for m in models if 'flash' in m.lower()), models[0] if models else 'gemini-pro')
+        model = genai.GenerativeModel(model_name)
         
         full_prompt = f"""Generate simple Manim code.