Update main.py

main.py

@@ -1,7 +1,7 @@
 import os
 import time
 import requests
-from fastapi import FastAPI, File, UploadFile
+from fastapi import FastAPI, File, UploadFile, Response
 from langchain_openai import ChatOpenAI
 from tempfile import NamedTemporaryFile
 
@@ -111,3 +111,17 @@ async def predict(file: UploadFile = File(...)):
     log_message(f"Overall /predict process took {overall_end - overall_start:.4f} seconds.")
 
     return {"prediction": prediction}
+
+@app.get("/logs")
+def get_logs():
+    """
+    Endpoint สำหรับดาวน์โหลด log file
+    """
+    log_file = "/app/logs/processing_logs.txt"  # เปลี่ยนให้ตรงกับ path ที่กำหนดใน logger.py
+    if not os.path.exists(log_file):
+        return {"error": "Log file not found"}
+    
+    with open(log_file, "r", encoding="utf-8") as f:
+        content = f.read()
+    
+    return Response(content, media_type="text/plain")