Update app.py

app.py

@@ -1,11 +1,8 @@
-from fastapi import FastAPI, File, UploadFile, HTTPException
-from fastapi.responses import JSONResponse
-from pydantic import BaseModel
+from flask import Flask, request, jsonify
 import shutil
 import os
-import json
 
-app = FastAPI(docs_url=None, redoc_url=None)
+app = Flask(__name__)
 
 # Create folders to store uploaded images
 UPLOAD_FOLDER = "uploads"
@@ -18,64 +15,67 @@ for folder in [UPLOAD_FOLDER, IMAGE_FOLDER, VIDEO_FOLDER]:
 # Create a dictionary to store available images
 available_images = {}
 
+@app.route("/upload", methods=["POST"])
+def upload_image():
+    url = request.form.get("url")
+    file = request.files["file"]
 
-class ImageInfo(BaseModel):
-    url: str
-    name: str
-
-
-@app.post("/upload")
-async def upload_image(url: str, file: UploadFile = File(...)):
     allowed_formats = [".png", ".jpg"]
     file_format = os.path.splitext(file.filename)[1]
 
     if file_format not in allowed_formats:
-        raise HTTPException(status_code=400, detail="Unsupported file format")
+        return jsonify({"message": "Unsupported file format"}), 400
 
     # Save the uploaded file
-    with open(os.path.join(UPLOAD_FOLDER, file.filename), "wb") as buffer:
-        shutil.copyfileobj(file.file, buffer)
+    file.save(os.path.join(UPLOAD_FOLDER, file.filename))
 
     # Determine the folder based on the file format
     folder = "images" if file_format in [".png", ".jpg"] else "videos"
 
     # Move the file to the appropriate folder
-    os.rename(
+    shutil.move(
         os.path.join(UPLOAD_FOLDER, file.filename),
         os.path.join(UPLOAD_FOLDER, folder, file.filename)
     )
 
     # Update available images dictionary
-    available_images[file.filename] = ImageInfo(url=url, name=file.filename)
+    available_images[file.filename] = {"url": url, "name": file.filename}
+
+    return jsonify({"message": "File uploaded successfully"})
 
-    return JSONResponse(content={"message": "File uploaded successfully"})
 
+@app.route("/get-all-images", methods=["GET"])
+def get_all_images():
+    password = request.args.get("password")
+    from_folder = request.args.get("from_folder")
 
-@app.get("/get-all-images")
-async def get_all_images(password: str, from_folder: str):
     if password != "Kastg@123":
-        raise HTTPException(status_code=401, detail="Unauthorized")
+        return jsonify({"message": "Unauthorized"}), 401
 
     folder_path = os.path.join(UPLOAD_FOLDER, from_folder)
     if not os.path.exists(folder_path):
-        raise HTTPException(status_code=404, detail="Folder not found")
+        return jsonify({"message": "Folder not found"}), 404
 
     images = os.listdir(folder_path)
-    return JSONResponse(content={"images": images})
+    return jsonify({"images": images})
 
 
-@app.delete("/delete-image")
-async def delete_image(image_name: str, password: str):
+@app.route("/delete-image", methods=["DELETE"])
+def delete_image():
+    image_name = request.args.get("image_name")
+    password = request.args.get("password")
+
     if password != "Kastg@123":
-        raise HTTPException(status_code=401, detail="Unauthorized")
+        return jsonify({"message": "Unauthorized"}), 401
 
     image_path = os.path.join(UPLOAD_FOLDER, image_name)
     if not os.path.exists(image_path):
-        raise HTTPException(status_code=404, detail="Image not found")
+        return jsonify({"message": "Image not found"}), 404
 
     os.remove(image_path)
     available_images.pop(image_name, None)
-    return JSONResponse(content={"message": "Image deleted successfully"})
+    return jsonify({"message": "Image deleted successfully"})
+
 
 if __name__ == "__main__":
-    app.run(host="0.0.0.0", port=7860, debug=True)
+    app.run(host="0.0.0.0", port=7860, debug=True)