no more FastAPI

app.py

@@ -1,13 +1,7 @@
 import sqlite3
-import time
-from pathlib import Path
 from sqlite3.dbapi2 import Connection
 
 import gradio as gr
-import uvicorn
-from fastapi import FastAPI
-from fastapi.middleware.cors import CORSMiddleware
-from fastapi.staticfiles import StaticFiles
 from PIL import Image, ImageDraw, ImageFont
 
 
@@ -21,7 +15,7 @@ def write_text_to_image(
     name_y: int = 410,
     distance_x: int = 435,
     distance_y: int = 450,
-) -> str:
+):
     # Save the input data to the SQLite database
     cursor = conn.cursor()
     cursor.execute(
@@ -36,7 +30,7 @@ def write_text_to_image(
     draw = ImageDraw.Draw(image)
 
     # Set the font (adjust the font size and style)
-    font = ImageFont.truetype("font/RobotoSlab-Bold.ttf", 25)
+    font = ImageFont.truetype("font/RobotoSlab-Medium.ttf", 25)
 
     # Define text colors
     text_color = (0, 255, 0)  # Green
@@ -47,36 +41,11 @@ def write_text_to_image(
     # Write the distance on the image at the specified coordinates
     draw.text((distance_x, distance_y), f"{distance} km", font=font, fill=text_color)
 
-    # Save the image with the text
-    timestamp = int(time.perf_counter_ns())
-    output_image_path = f"{static_dir}/{timestamp}.jpg"
-    image.save(output_image_path)
-
-    # Return the path to the generated image
-    return output_image_path
+    return image
 
 
 # Serve the app
 if __name__ == "__main__":
-    # Create a FastAPI app
-    app = FastAPI()
-
-    # Allow CORS (optional, but recommended for Hugging Face Spaces)
-    app.add_middleware(
-        CORSMiddleware,
-        allow_origins=["*"],
-        allow_credentials=True,
-        allow_methods=["*"],
-        allow_headers=["*"],
-    )
-
-    # Create a static directory to store the static files
-    static_dir = Path("./static")
-    static_dir.mkdir(parents=True, exist_ok=True)
-
-    # Mount FastAPI StaticFiles server
-    app.mount("/static", StaticFiles(directory=static_dir), name="static")
-
     # Connect to SQLite database
     conn = sqlite3.connect("user_input.db", check_same_thread=False)
     cursor = conn.cursor()
@@ -105,7 +74,7 @@ if __name__ == "__main__":
 
         # Event handler for the button click
         def on_submit(name, distance):
-            image_path = write_text_to_image(name, distance, conn)
+            image_path = write_text_to_image(name, distance, conn, static_dir="static")
             submit_button = gr.Button(visible=False)
             return submit_button, image_path
 
@@ -121,6 +90,4 @@ if __name__ == "__main__":
 
     # Mount Gradio app to FastAPI app
     demo.show_api = False
-    app = gr.mount_gradio_app(app, demo, path="/")
-
-    uvicorn.run(app, host="0.0.0.0", port=7860)
+    demo.launch()