Update app.py

app.py

@@ -1,4 +1,5 @@
 from fastapi import FastAPI, UploadFile, File, HTTPException
+from fastapi.responses import FileResponse
 from tensorflow.keras.models import load_model, Sequential
 from tensorflow.keras.layers import Dense, LSTM
 from tensorflow.keras.optimizers import Adam
@@ -65,8 +66,13 @@ async def retrain(model: UploadFile = File(...), data: str = None):
         updated_model_path = temp_model_path.replace(".h5", "_updated.h5")
         model.save(updated_model_path)
 
-        return {"message": "Model retrained successfully.", "updated_model_path": updated_model_path}
-
+        # Return the path for downloading
+        return FileResponse(
+            path=updated_model_path,
+            filename="updated_model.h5",
+            media_type="application/octet-stream",
+            headers={"Content-Disposition": "attachment; filename=updated_model.h5"}
+        )
     except Exception as e:
         raise HTTPException(status_code=500, detail=str(e))
     finally: