Update app.py

app.py

@@ -75,25 +75,51 @@ import requests
 import gradio as gr
 import os
 import json
+import ffmpeg
+
+# Function to convert video to audio using ffmpeg
+def convert_video_to_audio(video_path):
+    try:
+        output_path = f"flowly_ai_audio_converter{os.path.splittext(video_path)[0]}.mp3"
+        ffmpeg.input(video_path).output(output_path).run()
+        return output_path
+    except Exception as e:
+        return f"Error converting video: {str(e)}"
 
 # Function to recognize audio from URL or uploaded file
 def recognize_audio(choice, url, file):
     api_url = os.getenv("API_URL", "https://api.audd.io/").strip('"')
     params = {
         "return": "apple_music,spotify",
-        "api_token": "test"
+        "api_token": os.getenv("API_TOKEN")
     }
 
+    # Check if URL is provided
     if choice == "URL":
         if not url:
             return "Please enter a valid URL."
         params['url'] = url
         response = requests.post(api_url, data=params)
 
+    # Check if file is uploaded
     elif choice == "Upload File":
         if not file:
             return "Please upload a valid audio file."
-        with open(file, "rb") as f:
+
+        # Check if the uploaded file is a video (e.g., mp4)
+        file_extension = file.split('.')[-1].lower()
+        audio_file_path = file
+
+        video_formats = [data.upper() for data in (sorted(['3GP', 'ASF', 'AVI', 'DIVX', 'FLV', 'M2TS', 'M4V', 'MKV', 'MOV', 'MP4', 'MPEG', 'MPG', 'MTS', 'TS', 'VOB', 'WEBM', 'WMV', 'XVID'])) if data not in ['3GP', 'DIVX', 'XVID']]
+
+        if file_extension in video_formats:
+            # Convert video to audio file (mp3 format)
+            audio_file_path = convert_video_to_audio(file)
+            if audio_file_path.startswith("Error"):
+                return audio_file_path
+
+        # If it's already an audio file, use it as is
+        with open(audio_file_path, "rb") as f:
             response = requests.post(api_url, data=params, files={'file': f})
 
     else:
@@ -145,11 +171,11 @@ interface = gr.Interface(
     inputs=[
         gr.Radio(["URL", "Upload File"], label="Select Input Method"),
         gr.Textbox(label="Enter Audio URL", placeholder="https://example.com/audio.mp3"),
-        gr.File(label="Upload Audio File", type="filepath")  # Menggunakan filepath agar sesuai dengan Gradio
+        gr.File(label="Upload Audio or Video File", type="filepath")  # Menggunakan filepath agar sesuai dengan Gradio
     ],
     outputs=gr.Markdown(label="Recognition Result"),
     title="Audio Recognition",
-    description="Choose a method: Upload an audio file or enter a URL to identify the song."
+    description="Choose a method: Upload an audio/video file or enter a URL to identify the song."
 )
 
 # Run Gradio App