Update app.py

app.py

@@ -126,56 +126,56 @@ def askimage():
         return jsonify({"error": "No image provided"}), 400
     
     try:
-    # Read and compress image
-    image_bytes = image_file.read()
-    img = Image.open(BytesIO(image_bytes))
-    
-    # Convert RGBA to RGB if needed
-    if img.mode in ('RGBA', 'LA', 'P'):
-        background = Image.new('RGB', img.size, (255, 255, 255))
-        if img.mode == 'P':
-            img = img.convert('RGBA')
-        background.paste(img, mask=img.split()[-1] if img.mode == 'RGBA' else None)
-        img = background
-    
-    # Resize if too large (max 33 megapixels)
-    max_pixels = 33 * 1000000
-    if img.width * img.height > max_pixels:
-        ratio = (max_pixels / (img.width * img.height)) ** 0.5
-        new_size = (int(img.width * ratio), int(img.height * ratio))
-        img = img.resize(new_size, Image.Resampling.LANCZOS)
-    
-    # Compress to JPEG with quality adjustment to stay under 4MB
-    output = BytesIO()
-    quality = 85
-    while quality > 20:
-        output.seek(0)
-        output.truncate()
-        img.convert('RGB').save(output, format='JPEG', quality=quality, optimize=True)
-        size_mb = output.tell() / (1024 * 1024)
+        # Read and compress image
+        image_bytes = image_file.read()
+        img = Image.open(BytesIO(image_bytes))
         
-        # Base64 increases size by ~33%, so aim for 3MB max
-        if size_mb < 3.0:
-            break
-        quality -= 10
-    
-    compressed_bytes = output.getvalue()
-    base64_image = base64.b64encode(compressed_bytes).decode('utf-8')
-    
-    # Create data URL
-    image_data_url = f"data:image/jpeg;base64,{base64_image}"
-        # Build prompt in your existing format
-        prompt = "Image" + "%" + image_data_url + "%" + mode + "%" + lang
-        print(prompt)
+        # Convert RGBA to RGB if needed
+        if img.mode in ('RGBA', 'LA', 'P'):
+            background = Image.new('RGB', img.size, (255, 255, 255))
+            if img.mode == 'P':
+                img = img.convert('RGBA')
+            background.paste(img, mask=img.split()[-1] if img.mode == 'RGBA' else None)
+            img = background
         
-        return Response(
-            stream_with_context(stream_openrouter(prompt)),
-            mimetype="text/event-stream",
-            headers={
-                "Cache-Control": "no-cache",
-                "X-Accel-Buffering": "no"
-            }
-        )
+        # Resize if too large (max 33 megapixels)
+        max_pixels = 33 * 1000000
+        if img.width * img.height > max_pixels:
+            ratio = (max_pixels / (img.width * img.height)) ** 0.5
+            new_size = (int(img.width * ratio), int(img.height * ratio))
+            img = img.resize(new_size, Image.Resampling.LANCZOS)
+        
+        # Compress to JPEG with quality adjustment to stay under 4MB
+        output = BytesIO()
+        quality = 85
+        while quality > 20:
+            output.seek(0)
+            output.truncate()
+            img.convert('RGB').save(output, format='JPEG', quality=quality, optimize=True)
+            size_mb = output.tell() / (1024 * 1024)
+            
+            # Base64 increases size by ~33%, so aim for 3MB max
+            if size_mb < 3.0:
+                break
+            quality -= 10
+        
+        compressed_bytes = output.getvalue()
+        base64_image = base64.b64encode(compressed_bytes).decode('utf-8')
+    
+        # Create data URL
+        image_data_url = f"data:image/jpeg;base64,{base64_image}"
+            # Build prompt in your existing format
+            prompt = "Image" + "%" + image_data_url + "%" + mode + "%" + lang
+            print(prompt)
+            
+            return Response(
+                stream_with_context(stream_openrouter(prompt)),
+                mimetype="text/event-stream",
+                headers={
+                    "Cache-Control": "no-cache",
+                    "X-Accel-Buffering": "no"
+                }
+            )
     except Exception as e:
         return jsonify({"error": str(e)}), 500