Update app.py

app.py

@@ -8,44 +8,47 @@ app.secret_key = "peak_accuracy_trustify_key"
 UPLOAD_FOLDER = "uploads"
 os.makedirs(UPLOAD_FOLDER, exist_ok=True)
 app.config["UPLOAD_FOLDER"] = UPLOAD_FOLDER
-app.config["MAX_CONTENT_LENGTH"] = 100 * 1024 * 1024 # 100MB Limit
+app.config["MAX_CONTENT_LENGTH"] = 100 * 1024 * 1024 
 
 @app.route("/", methods=["GET", "POST"])
 def index():
     if request.method == "POST":
         content_type = request.form.get("type")
-
-        # Text Verification Route
+        
         if content_type == "text":
             text = request.form.get("text", "").strip()
             if not text:
                 flash("Please enter some text to verify.")
                 return redirect(url_for("index"))
-                
+            
             from services.text_checker import check_text
             result = check_text(text)
+            # Ensure your HTML file is named 'result.html' in the templates folder
             return render_template("result.html", result=result)
 
-        # Media Verification Route
         if content_type in ["image", "video"]:
             file = request.files.get("file")
             if not file or file.filename == "":
                 flash("No file selected.")
                 return redirect(url_for("index"))
-
+            
             filename = f"{uuid.uuid4()}_{secure_filename(file.filename)}"
             path = os.path.join(app.config["UPLOAD_FOLDER"], filename)
             file.save(path)
-
-            if content_type == "image":
-                from services.image_checker import check_image_hybrid
-                result = check_image_hybrid(path)
-            else:
-                from services.video_checker import check_video_authenticity
-                result = check_video_authenticity(path)
-
+            
+            try:
+                if content_type == "image":
+                    from services.image_checker import check_image_hybrid
+                    result = check_image_hybrid(path)
+                else:
+                    from services.video_checker import check_video_authenticity
+                    result = check_video_authenticity(path)
+            finally:
+                if os.path.exists(path):
+                    os.remove(path)
+                    
             return render_template("result.html", result=result)
-
+            
     return render_template("index.html")
 
 if __name__ == "__main__":