Update main.py

main.py

@@ -1,23 +1,20 @@
-from fastapi import FastAPI, File, UploadFile
-from fastapi import FastAPI, File, UploadFile, Form, Request
-from fastapi.responses import HTMLResponse, FileResponse
+from fastapi import FastAPI, File, UploadFile, HTTPException, Request
+from fastapi.responses import HTMLResponse, FileResponse, JSONResponse
 from fastapi.staticfiles import StaticFiles
 from fastapi.templating import Jinja2Templates
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import JSONResponse
-from fastapi.responses import StreamingResponse
-from gradio_client import Client
+from fastapi.middleware.cors import CORSMiddleware
+from gradio_client import Client, handle_file
 import os
-import io
 import tempfile
+import base64
+
 app = FastAPI()
 
+# Retrieve Hugging Face token from environment variables
 hf_token = os.environ.get('HF_TOKEN')
-client = Client("https://ashrafb-image-to-sketch.hf.space/", hf_token=hf_token)
-
-import tempfile
-from fastapi.middleware.cors import CORSMiddleware
+client = Client("Ashrafb/image-to-sketch-ari", hf_token=hf_token)
 
+# Configure CORS middleware
 app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],  # Adjust as needed, '*' allows requests from any origin
@@ -26,13 +23,6 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-
-import base64
-
-import logging
-
-logging.basicConfig(level=logging.DEBUG)
-
 @app.post("/upload/")
 async def upload_file(file: UploadFile = File(...)):
     # Save the uploaded file to a temporary location
@@ -65,15 +55,3 @@ async def upload_file(file: UploadFile = File(...)):
         # Clean up the temporary file
         if os.path.exists(temp_file_path):
             os.unlink(temp_file_path)
-
-# Serve static files from the "static" directory
-app.mount("/static", StaticFiles(directory="static"), name="static")
-
-@app.get("/", response_class=HTMLResponse)
-def index(request: Request):
-    return FileResponse("static/index.html")
-
-# Endpoint to serve the result files
-@app.get("/result/{filename}")
-async def get_result(filename: str):
-    return FileResponse(filename)