July 18 commit to Hugging Face 3:47pm

app.py

@@ -5,6 +5,34 @@ import time
 import numpy as np
 import json
 
+from fastapi import FastAPI, HTTPException
+from fastapi.responses import FileResponse, JSONResponse
+
+app = FastAPI()
+
+#--------------------------------------------------------------------------------
+@app.get("/list-files")
+def list_files(directory: str = ".", extension: str = ".txt"):
+    if not os.path.exists(directory):
+        raise HTTPException(status_code=404, detail="Directory not found")
+
+    files = [f for f in os.listdir(directory) if f.endswith(extension)]
+    return JSONResponse(content={"files": files})
+
+@app.get("/download-file/{filename}")
+def download_file(filename: str, directory: str = "."):
+    file_path = os.path.join(directory, filename)
+    if not os.path.exists(file_path):
+        raise HTTPException(status_code=404, detail="File not found")
+    return FileResponse(path=file_path, filename=filename)
+
+def run_fastapi():
+    import uvicorn
+    uvicorn.run(app, host="0.0.0.0", port=8000)
+
+#--------------------------------------------------------------------------------
+
+
 qs = {}
 
 theme = gr.themes.Base(
@@ -290,7 +318,11 @@ def log_message(param, session_id):
 # Launching
 # --------------------------------------------------------------------------------------------
 
+import threading
+
 app_inits()    
+
+threading.Thread(target=run_fastapi).start()
 demo.queue()
 demo.launch(allowed_paths=["."], max_threads=40) 
 

fe5ad737-4460-4d52-97d8-5895668d9920_log_data.txt

@@ -0,0 +1,2 @@
+2024-07-18 15:43:36: : -ENTRY- [query]: debug [params]  [keyword] None
+2024-07-18 15:43:36: : -EXIT- [query]: debug [params]  [keyword] None