Update app.py

app.py

@@ -31,6 +31,7 @@ app.add_middleware(
     CORSMiddleware,
     allow_origins=[
         "https://hivili.web.app",
+        "https://hivili.com",
         "http://localhost:3000",
         "https://*.lovable.dev",
         "https://*.sandbox.lovable.dev",
@@ -110,7 +111,7 @@ def validate_image(image_content: bytes):
         raise HTTPException(status_code=400, detail=f"Image validation error: {str(e)}")
 
 def generate_gemini_image(images: List[Image.Image], prompt: str):
-    """Generate or edit image using Gemini API"""
+    """Generate or edit image using Gemini API and resize to 1024x1024 pixels"""
     try:
         contents = images + [prompt]
         response = genai.GenerativeModel(GEMINI_MODEL).generate_content(contents)
@@ -125,9 +126,15 @@ def generate_gemini_image(images: List[Image.Image], prompt: str):
                 with tempfile.NamedTemporaryFile(suffix=".png", delete=False) as tmp:
                     temp_path = tmp.name
                     generated_image = Image.open(BytesIO(part.inline_data.data))
-                    generated_image.save(temp_path)
+                    # Resize image to 1024x1024 pixels
+                    generated_image = generated_image.resize((1024, 1024), Image.LANCZOS)
+                    # Verify dimensions
+                    if generated_image.size != (1024, 1024):
+                        logger.error(f"Resized image dimensions {generated_image.size} do not match expected 1024x1024")
+                        raise HTTPException(status_code=500, detail="Failed to resize image to 1024x1024")
+                    generated_image.save(temp_path, format="PNG")
                     image_path = temp_path
-                    logger.info(f"Generated image saved to: {temp_path} with prompt: {prompt}")
+                    logger.info(f"Generated and resized image to 1024x1024 saved to: {temp_path} with prompt: {prompt}")
 
         if image_path:
             return image_path, ""