Upload main.py

main.py

@@ -46,7 +46,7 @@ from telegraph import Telegraph, upload_file
 from pathlib import Path
 from serpapi import GoogleSearch
 
-from fastapi import FastAPI, UploadFile, File
+from fastapi import FastAPI, UploadFile, File, Response
 from fastapi.security import OAuth2PasswordBearer, OAuth2PasswordRequestForm
 from fastapi import Depends, FastAPI, HTTPException, status
 from fastapi.openapi.utils import get_openapi
@@ -377,17 +377,24 @@ def ryuzaki_ai(
         return {"status": "false", "message": "Internal server error"}
 
 @app.get("/ryuzaki/unsplash")
-async def get_image_unsplash(query: str, size: str="500x500"):
+async def get_image_unsplash(query: str, size: str = "500x500"):
     url = SOURCE_UNSPLASH_URL
     image_url = f"{url}/?{query}/{size}"
-
     try:
         response = requests.get(image_url)
         response.raise_for_status()
-    except requests.exceptions.RequestException:
-        raise HTTPException(status_code=500, detail="Internal server error")
-
-    return StreamingResponse(BytesIO(response.content), media_type="image/jpeg")
+    except requests.exceptions.RequestException as e:
+        raise HTTPException(status_code=500, detail=f"Internal server error: {str(e)}")
+    try:
+        encoded_string = base64.b64encode(response.content).decode("utf-8")
+    except Exception as e:
+        raise HTTPException(status_code=500, detail=f"Error encoding image: {str(e)}")
+    headers = {"Content-Type": "image/jpeg"}
+    return Response(
+        content=encoded_string,
+        media_type="image/jpeg",
+        headers=headers
+    )
 
 @app.get("/ryuzaki/calligraphy")
 async def get_image_calligraphy(query: str):