File size: 1,501 Bytes
d69b7ce ee3e372 e482b3f ee3e372 d69b7ce ee3e372 d69b7ce afaca69 ee3e372 afaca69 d69b7ce ee3e372 d69b7ce ee3e372 d69b7ce ee3e372 d69b7ce |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 |
from fastapi import FastAPI, File, UploadFile, HTTPException
from fastapi.responses import JSONResponse
from gradio_client import Client, handle_file
import os
import base64
# Initialize FastAPI
app = FastAPI()
# Load environment variables if needed
HF_TOKEN = os.getenv("HF_TOKEN")
# Initialize the Gradio client with the token
client = Client("Makhinur/Its", hf_token=HF_TOKEN)
@app.post("/upload/")
async def upload_image(file: UploadFile = File(...)):
try:
# Save the uploaded file to a temporary location
file_location = f"temp_{file.filename}"
with open(file_location, "wb") as f:
f.write(await file.read())
# Use the Gradio client to send the image to the Gradio app
result = client.predict(
img=handle_file(file_location),
api_name="/predict"
)
# Clean up the temporary file
os.remove(file_location)
# Read the sketch image from the result
sketch_image_path = result[0] # Path to the sketch image
# Convert the sketch image to base64
with open(sketch_image_path, "rb") as img_file:
sketch_image_base64 = base64.b64encode(img_file.read()).decode('utf-8')
# Prepare the response
response = {
"sketch_image_base64": f"data:image/jpeg;base64,{sketch_image_base64}"
}
return JSONResponse(content=response)
except Exception as e:
raise HTTPException(status_code=500, detail=str(e))
|