logging added

app/main.py

@@ -7,6 +7,8 @@ from pydantic import BaseModel
 import subprocess
 from pathlib import Path
 from config import settings
+import logging
+
 
 app = FastAPI(
     title=settings.PROJECT_NAME, openapi_url=f"{settings.API_V1_STR}/openapi.json"
@@ -27,29 +29,33 @@ async def root(request: Request):
 
 @app.post("/detect", response_class=HTMLResponse)
 async def detect_objects(request: Request, item: Item):
-    print('Item:', item)
+    try:
+        print('Item:', item)
 
-    print('File name:', item.file.filename)
-    
-    input_file = f"uploads/{item.file.filename}"
-    output_file = f"static/output/{item.file.filename}"
+        print('File name:', item.file.filename)
+        
+        input_file = f"uploads/{item.file.filename}"
+        output_file = f"static/output/{item.file.filename}"
 
-    # Save the uploaded file
-    with open(input_file, "wb") as file:
-        file.write(item.file.file.read())
+        # Save the uploaded file
+        with open(input_file, "wb") as file:
+            file.write(item.file.file.read())
 
-    print('Detect start')
+        print('Detect start')
 
-    # Run YOLOv7 detection and save output
-    subprocess.run(["python", "detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt",  "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--project", output_file])
+        # Run YOLOv7 detection and save output
+        subprocess.run(["python", "detect.py", "--conf", "0.5", "--img-size", "640", "--weights", "app/model/best.pt",  "--source", input_file, "--save-txt", "--save-conf", "--exist-ok", "--project", output_file])
 
-    print('Detect end')
+        print('Detect end')
 
-    # Render HTML using Jinja2Templates
-    return templates.TemplateResponse(
-        "result.html",
-        {"request": request, "original_image": f"/static/{item.file.filename}", "output_image": f"/static/output/{item.file.filename}"},
-    )
+        # Render HTML using Jinja2Templates
+        return templates.TemplateResponse(
+            "result.html",
+            {"request": request, "original_image": f"/static/{item.file.filename}", "output_image": f"/static/output/{item.file.filename}"},
+        )
+    except Exception as e:
+        logging.error(f"Error in /detect endpoint: {str(e)}")
+        raise e