Update app.py

app.py

@@ -6,6 +6,15 @@ import glob
 import shutil
 import gradio as gr
 
+allowed_extensions = [
+    ".3g2", ".3gp", ".3gpp", ".avi", ".cavs", ".dv", ".dvr", ".flv", ".m2ts",
+    ".m4v", ".mkv", ".mod", ".mov", ".mp4", ".mpeg", ".mpg", ".mts", ".mxf",
+    ".ogg", ".rm", ".rmvb", ".swf", ".ts", ".vob", ".webm", ".wmv", ".wtv",
+    ".ogv", ".opus", ".aac", ".ac3", ".aif", ".aifc", ".aiff", ".amr", ".au",
+    ".caf", ".dss", ".flac", ".m4a", ".m4b", ".mp3", ".oga", ".voc", ".wav",
+    ".weba", ".wma"
+]
+
 os.system("chmod +x fdkaac")  # Make sure fdkaac is executable
 
 accel = 'auto'
@@ -123,7 +132,9 @@ async def process_conversion(use_youtube, youtube_url, video_file, downscale, fa
         else:
             if not video_file:
                 return "No video provided.", None
-            ext = os.path.splitext(video_file.name)[1]
+            ext = os.path.splitext(video_file.name)[1].lower()
+            if ext not in allowed_extensions:
+                return f"Unsupported file type: {ext}", None
             input_path = os.path.join(UPLOAD_FOLDER, f"{uuid.uuid4()}{ext}")
             shutil.copy2(video_file.name, input_path)
 
@@ -152,7 +163,7 @@ with gr.Blocks(theme=gr.themes.Default(primary_hue="rose")) as demo:
         with gr.Row():
             use_youtube = gr.Checkbox(label="🔗 Use YouTube URL (usually broken here)", value=False)
             youtube_url = gr.Textbox(label="YouTube URL", placeholder="Paste YouTube URL here")
-        video_file = gr.File(label="📁 Upload Video File", file_types=["video/*", "audio/*"])
+        video_file = gr.File(label="📁 Upload Video File", file_types=[".3g2", ".3gp", ".3gpp", ".avi", ".cavs", ".dv", ".dvr", ".flv", ".m2ts", ".m4v", ".mkv", ".mod", ".mov", ".mp4", ".mpeg", ".mpg", ".mts", ".mxf", ".ogg", ".rm", ".rmvb", ".swf", ".ts", ".vob", ".webm", ".wmv", ".wtv", ".ogv", ".opus", ".aac", ".ac3", ".aif", ".aifc", ".aiff", ".amr", ".au", ".caf", ".dss", ".flac", ".m4a", ".m4b", ".mp3", ".oga", ".voc", ".wav", ".weba", ".wma"])
 
     gr.Markdown("### ⚙️ **Conversion Settings**")
     with gr.Group():