Update app.py

app.py

@@ -3,6 +3,7 @@ import os
 import sys
 
 from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import RedirectResponse
 import gradio as gr
 import requests
 from typing import List
@@ -41,7 +42,7 @@ ds = []
 images = []
 
 @app.post("/index")
-async def index(files: List[UploadFile] = File(...)):
+def index(files: List[UploadFile] = File(...)):
     global ds, images
     images = []
     ds = []
@@ -65,7 +66,7 @@ async def index(files: List[UploadFile] = File(...)):
     return {"message": f"Uploaded and converted {len(images)} pages"}
 
 @app.post("/search")
-async def search(query: str, k: int):
+def search(query: str, k: int):
     qs = []
     with torch.no_grad():
         batch_query = process_queries(processor, [query], mock_image)
@@ -80,4 +81,9 @@ async def search(query: str, k: int):
 
     results = [{"page": idx, "image": "image_placeholder"} for idx in top_k_indices]
 
-    return {"results": results}
+    return {"results": results}
+
+# Rediriger la racine vers /docs
+@app.get("/")
+def read_root():
+    return RedirectResponse(url="/docs")