Update app.py

app.py

@@ -71,11 +71,17 @@
 # # Launch the Gradio interface
 # iface.launch()
 
-import requests
-import gradio as gr
 import os
 import json
+import requests
 import ffmpeg
+import gradio as gr
+import tempfile
+import threading
+import time
+
+# Video formats that are supported
+video_formats = ['3gp', 'asf', 'avi', 'divx', 'flv', 'm2ts', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'mts', 'ts', 'vob', 'webm', 'wmv', 'xvid']
 
 # Function to convert video to audio using ffmpeg
 def convert_video_to_audio(video_path):
@@ -93,9 +99,25 @@ def convert_video_to_audio(video_path):
     except Exception as e:
         return f"Error converting video: {str(e)}"
 
+# Function to download file from URL
+def download_file(url):
+    try:
+        response = requests.get(url, stream=True)
+        filename = os.path.join("temp_downloads", url.split("/")[-1])  # Save in temp_downloads folder
+        os.makedirs("temp_downloads", exist_ok=True)
+
+        # Save file content
+        with open(filename, 'wb') as f:
+            for chunk in response.iter_content(chunk_size=8192):
+                f.write(chunk)
+
+        return filename
+    except Exception as e:
+        return f"Error downloading file: {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('"')
+    api_url = os.getenv("API_URL")
     params = {
         "return": "apple_music,spotify",
         "api_token": os.getenv("API_TOKEN")
@@ -105,8 +127,22 @@ def recognize_audio(choice, url, file):
     if choice == "URL":
         if not url:
             return "Please enter a valid URL."
-        params['url'] = url
-        response = requests.post(api_url, data=params)
+        
+        # Check if the URL ends with a video file extension
+        url_extension = url.split('.')[-1].lower()
+
+        if url_extension in video_formats:
+            # Download video and process as video
+            video_path = download_file(url)
+            if video_path.startswith("Error"):
+                return video_path
+            audio_file_path = convert_video_to_audio(video_path)
+            if audio_file_path.startswith("Error"):
+                return audio_file_path
+        else:
+            # Process URL as audio
+            params['url'] = url
+            response = requests.post(api_url, data=params)
 
     # Check if file is uploaded
     elif choice == "Upload File":
@@ -117,8 +153,6 @@ def recognize_audio(choice, url, file):
         file_extension = file.split('.')[-1].lower()
         audio_file_path = file
 
-        video_formats = ['3gp', 'asf', 'avi', 'divx', 'flv', 'm2ts', 'm4v', 'mkv', 'mov', 'mp4', 'mpeg', 'mpg', 'mts', 'ts', 'vob', 'webm', 'wmv', 'xvid']
-
         if file_extension in video_formats:
             # Convert video to audio file (mp3 format)
             audio_file_path = convert_video_to_audio(file)
@@ -180,12 +214,23 @@ def recognize_audio(choice, url, file):
     except Exception as e:
         return f"Error parsing response: {str(e)}"
 
+# Function to cache the audio file and delete it after 15 minutes
+def cache_and_delete_file(audio_file_path):
+    # Wait for 15 minutes before deleting the file
+    time.sleep(900)  # 900 seconds = 15 minutes
+    
+    try:
+        os.remove(audio_file_path)
+        print(f"File {audio_file_path} has been deleted after 15 minutes.")
+    except Exception as e:
+        print(f"Error deleting file {audio_file_path}: {str(e)}")
+
 # Gradio Interface
 interface = gr.Interface(
     fn=recognize_audio,
     inputs=[
         gr.Radio(["URL", "Upload File"], label="Select Input Method"),
-        gr.Textbox(label="Enter Audio URL", placeholder="https://example.com/audio.mp3"),
+        gr.Textbox(label="Enter Audio/Video URL", placeholder="https://example.com/audio.mp3 or video.mp4"),
         gr.File(label="Upload Audio or Video File", type="filepath")  # Menggunakan filepath agar sesuai dengan Gradio
     ],
     outputs=gr.Markdown(label="Recognition Result"),
@@ -195,4 +240,5 @@ interface = gr.Interface(
 
 # Run Gradio App
 if __name__ == "__main__":
+    # Launch the Gradio interface
     interface.launch()