Update app.py

app.py

@@ -21,9 +21,8 @@ download_models()
 
 @app.route('/detect', methods=['POST'])
 def detect_deepfake():
-    data_type = request.form.get('type')  # "audio"
+    # Expect an audio file in the request
     audio_file = request.files.get('audio_file')
-    folder_path = request.form.get('folder_path')
 
     # If a single audio file is provided
     if audio_file:
@@ -44,28 +43,9 @@ def detect_deepfake():
         except Exception as e:
             return jsonify({"error": str(e)}), 500
 
-    # If a folder path is provided
-    elif folder_path and os.path.isdir(folder_path):
-        results = {}
-        try:
-            for file_name in os.listdir(folder_path):
-                if file_name.endswith('.wav') or file_name.endswith('.mp3'):
-                    file_path = os.path.join(folder_path, file_name)
-                    result = audio_model(file_path)
-                    results[file_name] = {item['label']: item['score'] for item in result}
-
-            # Save results to a file
-            with open('detection_results.json', 'w') as f:
-                f.write(json.dumps(results))
-
-            return jsonify({"message": "Detection completed", "results": results}), 200
-
-        except Exception as e:
-            return jsonify({"error": str(e)}), 500
-
-    # Invalid request if neither audio file nor valid folder path is provided
+    # Invalid request if no audio file is provided
     else:
-        return jsonify({"error": "Invalid input. Provide an audio file or a valid folder path."}), 400
+        return jsonify({"error": "Invalid input. Please provide an audio file."}), 400
 
 if __name__ == '__main__':
     # Run the Flask app