Update main.py

main.py

@@ -24,6 +24,17 @@ CODE_TO_PLANT = {
 @app.get("/health")
 async def api_health_check():
     return JSONResponse(content={"status": "Service is running"})
+    
+@app.get("/download/{plant_name}")
+async def download_model(plant_name: str):
+    filename = f"model_{plant_name}.tflite"
+    file_path = os.path.join(MODEL_DIRECTORY, filename)
+
+    # Check if file exists
+    if not os.path.isfile(file_path):
+        raise HTTPException(status_code=404, detail=f"Model file '{filename}' not found")
+
+    return FileResponse(file_path, filename=filename, media_type="application/octet-stream")
 
 @app.get("/download_by_code/{code}")
 async def download_model_by_code(code: str):
@@ -41,3 +52,5 @@ async def download_model_by_code(code: str):
 
 if __name__ == "__main__":
     uvicorn.run(app, host="0.0.0.0", port=7860)
+
+