imagegpt / app.py
zliang's picture
Update app.py
588da92 verified
raw
history blame
713 Bytes
from fastapi import FastAPI
import base64
import os
import sys
import uvicorn
from imagekitio import ImageKit
import base64
app = FastAPI()
imagekit = ImageKit(
public_key=os.environ.get("public_key"),
private_key=os.environ.get("private_key"),
url_endpoint=os.environ.get("url_endpoint")
)
@app.get("/")
def upload_image(url):
try:
# Upload the image to ImageKit
upload = imagekit.upload(
file=url,
file_name="image.jpg",
)
# Return the upload response
return upload
except Exception as e:
return {"error": str(e)}
if __name__ == "__main__":
import uvicorn
uvicorn.run(app, host="0.0.0.0", port=7860)