Update models.py

models.py

@@ -71,7 +71,7 @@ def get_generation_prompt(
             # Contents should be images first, then the text prompt
             contents = images + [analysis_prompt]
             response = GEMINI_CLIENT.models.generate_content(
-                model='gemini-2.5-flash',
+                model='Gemini-2.0-flash',
                 contents=contents
             )
             expanded_prompt = response.text.strip()
@@ -86,7 +86,7 @@ def get_generation_prompt(
         if not OPENAI_CLIENT:
             return f"OpenAI API Key missing or client failed to initialize. Fallback prompt: Fusion of provided visual elements inspired by the prompt: {prompt}."
         try:
-            # Prepare contents for GPT-4o with base64 encoded images
+            # Prepare contents for gpt-image-1-low with base64 encoded images
             contents = [
                 {"type": "text", "text": analysis_prompt}
             ]
@@ -104,18 +104,18 @@ def get_generation_prompt(
                 })
 
             response = OPENAI_CLIENT.chat.completions.create(
-                model="gpt-4o",
+                model="gpt-image-1-low",
                 messages=[
                     {"role": "user", "content": contents}
                 ],
                 max_tokens=500
             )
             expanded_prompt = response.choices[0].message.content.strip()
-            print(f"GPT-4o Analysis Output: {expanded_prompt}")
+            print(f"gpt-image-1-low Analysis Output: {expanded_prompt}")
             return expanded_prompt
         except Exception as e:
             print(f"GPT API Error: {e}")
-            return f"Error using GPT-4o for analysis. Fallback prompt: Creative fusion of the three elements provided, inspired by the theme: {prompt}."
+            return f"Error using gpt-image-1-low for analysis. Fallback prompt: Creative fusion of the three elements provided, inspired by the theme: {prompt}."
     
     # Fallback if model is unrecognized
     return f"Creative synthesis of the visual elements provided, inspired by the prompt: {prompt}. Ensure photorealistic quality."