api

backend/app.py

@@ -62,6 +62,8 @@ if os.path.exists(frontend_path):
 else:
     print(f"Warning: Frontend build directory '{frontend_path}' does not exist.")
 
+api = FastAPI()
+
 def overlay_mask(base_image, mask_image, color_idx):
     """
     Given the base_image and the 0/1 mask, overlay the mask with the color indexed by the color_idx.
@@ -150,7 +152,7 @@ async def return_state_data(state):
     response['mask_overlayed_image'] = base64_str
     return response
 
-@app.get("/return_thumbnails")
+@api.get("/return_thumbnails")
 async def return_thumbnails_endpoint():
     thumbnails = await return_thumbnails()
     encoded_images = []
@@ -161,7 +163,7 @@ async def return_thumbnails_endpoint():
         encoded_images.append(base64_str)
     return JSONResponse(content={"thumbnails": encoded_images})
 
-@app.get("/return_state_data")
+@api.get("/return_state_data")
 async def return_state_data_endpoint(
     image_index: int = Query(...),
     detail_level: int = Query(...),
@@ -181,6 +183,8 @@ async def return_state_data_endpoint(
     response = await return_state_data(state)
     return response
 
+app.mount("/api", api)
+
 if __name__ == "__main__":
     import uvicorn
     uvicorn.run(app, host="0.0.0.0", port=7860)

frontend/src/App.js

@@ -123,7 +123,7 @@ const App = () => {
   useEffect(() => {
     const fetchThumbnails = async () => {
       try {
-        const response = await axios.get('/return_thumbnails');
+        const response = await axios.get('/api/return_thumbnails');
         const { thumbnails } = response.data;
         setThumbnails(thumbnails.map((base64) => `data:image/png;base64,${base64}`)); // Decode Base64
       } catch (error) {
@@ -144,7 +144,7 @@ const App = () => {
       // join the valid indices with commas
       const validIndicesStr = selectedValidIndices.length === 0 ? 'None' : selectedValidIndices.join(',');
       
-      const response = await axios.get('/return_state_data', {
+      const response = await axios.get('/api/return_state_data', {
         params: {
           image_index: selectedIdx,
           detail_level: selectedDot,