new endpoint

main.py

@@ -4,6 +4,7 @@ from io import BytesIO
 import torch
 from fastapi import FastAPI, Query
 from PIL import Image
+from fastapi import FastAPI, File, UploadFile, HTTPException
 from qwen_vl_utils import process_vision_info
 from transformers import AutoProcessor, Qwen2_5_VLForConditionalGeneration, Qwen2VLForConditionalGeneration
 
@@ -28,35 +29,61 @@ def read_root():
     return {"message": "API is live. Use the /predict endpoint."}
 
 
-def encode_image(image_path, max_size=(800, 800), quality=85):
+# def encode_image(image_path, max_size=(800, 800), quality=85):
+#     """
+#     Converts an image from a local file path to a Base64-encoded string with optimized size.
+
+#     Args:
+#         image_path (str): The path to the image file.
+#         max_size (tuple): The maximum width and height of the resized image.
+#         quality (int): The compression quality (1-100, higher means better quality but bigger size).
+
+#     Returns:
+#         str: Base64-encoded representation of the optimized image.
+#     """
+#     try:
+#         with Image.open(image_path) as img:
+#             # Convert to RGB (avoid issues with PNG transparency)
+#             img = img.convert("RGB")
+
+#             # Resize while maintaining aspect ratio
+#             img.thumbnail(max_size, Image.LANCZOS)
+
+#             # Save to buffer with compression
+#             buffer = BytesIO()
+#             img.save(
+#                 buffer, format="JPEG", quality=quality
+#             )  # Save as JPEG to reduce size
+#             return base64.b64encode(buffer.getvalue()).decode("utf-8")
+#     except Exception as e:
+#         print(f"❌ Error encoding image {image_path}: {e}")
+#         return None
+
+def encode_image(image_data: BytesIO, max_size=(800, 800), quality=85):
     """
-    Converts an image from a local file path to a Base64-encoded string with optimized size.
-
-    Args:
-        image_path (str): The path to the image file.
-        max_size (tuple): The maximum width and height of the resized image.
-        quality (int): The compression quality (1-100, higher means better quality but bigger size).
-
-    Returns:
-        str: Base64-encoded representation of the optimized image.
+    Converts an image from file data to a Base64-encoded string with optimized size.
     """
     try:
-        with Image.open(image_path) as img:
-            # Convert to RGB (avoid issues with PNG transparency)
+        with Image.open(image_data) as img:
             img = img.convert("RGB")
-
-            # Resize while maintaining aspect ratio
             img.thumbnail(max_size, Image.LANCZOS)
-
-            # Save to buffer with compression
             buffer = BytesIO()
-            img.save(
-                buffer, format="JPEG", quality=quality
-            )  # Save as JPEG to reduce size
+            img.save(buffer, format="JPEG", quality=quality)
             return base64.b64encode(buffer.getvalue()).decode("utf-8")
     except Exception as e:
-        print(f"❌ Error encoding image {image_path}: {e}")
-        return None
+        raise HTTPException(status_code=500, detail=f"Error encoding image: {e}")
+
+@app.post("/encode-image/")
+async def upload_and_encode_image(file: UploadFile = File(...)):
+    """
+    Endpoint to upload an image file and return its Base64-encoded representation.
+    """
+    try:
+        image_data = BytesIO(await file.read())
+        encoded_string = encode_image(image_data)
+        return {"filename": file.filename, "encoded_image": encoded_string}
+    except Exception as e:
+        raise HTTPException(status_code=400, detail=f"Invalid file: {e}")
 
 @app.get("/predict")
 def describe_image_with_qwen2_vl(image_url: str = Query(...), prompt: str = Query(...)):