Update app.py

app.py

@@ -1,26 +1,38 @@
-from fastapi import FastAPI, UploadFile, File, HTTPException
-import os
-import google.generativeai as genai
-from PIL import Image
-from io import BytesIO
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import StreamingResponse
+from PIL import Image, ImageOps, ImageFilter
+import io
+import numpy as np
 
 app = FastAPI()
 
-GOOGLE_API_KEY = os.getenv('GOOGLE_API_KEY')
+@app.post("/")
+async def convert_to_sketch(file: UploadFile = File(...)):
+    # Load the image
+    image = Image.open(io.BytesIO(await file.read()))
 
-if not GOOGLE_API_KEY:
-    raise ValueError("GOOGLE_API_KEY environment variable not set")
+    # Convert to grayscale
+    grayscale = ImageOps.grayscale(image)
 
-genai.configure(api_key=GOOGLE_API_KEY)
-model = genai.GenerativeModel('gemini-1.5-flash')
+    # Invert the grayscale image
+    inverted = ImageOps.invert(grayscale)
 
-@app.post("/generate-caption/")
-async def generate_caption(file: UploadFile = File(...)):
-    try:
-        image_data = await file.read()
-        image = Image.open(BytesIO(image_data))
-        
-        response = model.generate_content(image)
-        return {"caption": response.text}
-    except Exception as e:
-        raise HTTPException(status_code=500, detail=str(e))
+    # Apply Gaussian blur to the inverted image
+    blurred = inverted.filter(ImageFilter.GaussianBlur(15))
+
+    # Blend the grayscale image with the blurred inverted image using color dodge
+    blended = Image.blend(grayscale, blurred, 0.5)
+
+    # Enhance the contrast (optional)
+    enhanced = ImageOps.autocontrast(blended, cutoff=2)
+
+    # Save the resulting image to a bytes buffer
+    buf = io.BytesIO()
+    enhanced.save(buf, format='PNG')
+    buf.seek(0)
+
+    return StreamingResponse(buf, media_type="image/png")
+
+if __name__ == "__main__":
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)