Update app.py

app.py

@@ -2,43 +2,98 @@ import gradio as gr
 import yt_dlp
 import os
 from pydub import AudioSegment
+from redis import Redis
+import re
+import requests
+import io
+
+# Set up Redis connection
+redis = Redis(
+    host="amused-walleye-31373.upstash.io",
+    port=31373,
+    password="67212443600c45ffa962e19ea9202605",
+    ssl=True,
+    db=0
+)
+
+# Cloudflare Worker URL
+worker_base_url = "https://server.jessejesse.workers.dev"  # Your Cloudflare Worker URL
 
 # Ensure downloads directory exists
 os.makedirs("downloads", exist_ok=True)
 
+def sanitize_filename(filename):
+    """Sanitize the filename by removing or replacing special characters."""
+    return re.sub(r'[^a-zA-Z0-9_-]', '_', filename)
+
+def save_ringtone_to_worker(song_name, file_path):
+    """Upload the ringtone to Cloudflare Worker storage (R2 or KV) and save its URL in Redis."""
+    try:
+        sanitized_song_name = sanitize_filename(song_name)
+        
+        # Read file contents
+        with open(file_path, 'rb') as file:
+            file_data = file.read()
+        
+        # Upload file to the worker storage (e.g., R2 or KV)
+        response = requests.put(
+            f"{worker_base_url}/{sanitized_song_name}",
+            data=file_data,
+            headers={"Content-Type": "audio/mpeg"}  # For MP3, adjust for other formats
+        )
+        
+        if response.status_code == 200:
+            file_url = f"{worker_base_url}/{sanitized_song_name}"
+            redis.set(sanitized_song_name, file_url)  # Save the URL to Redis
+            print(f"Saved {sanitized_song_name} URL to Redis: {file_url}")
+        else:
+            print(f"Failed to upload file. Status code: {response.status_code}")
+        
+    except Exception as e:
+        print(f"Error uploading to Worker storage: {e}")
+
 def process_youtube_url(url, uploaded_file):
     try:
         filename = None
+        song_name = None
 
         if url:
             ydl_opts = {
                 'format': 'bestaudio/best',
                 'outtmpl': 'downloads/%(id)s.%(ext)s',
-                'cookiefile': 'cookies.txt'  # YTcookies 
+                'cookiefile': 'cookies.txt'  # YT cookies (if needed)
             }
 
             with yt_dlp.YoutubeDL(ydl_opts) as ydl:
                 info = ydl.extract_info(url, download=True)
                 filename = os.path.join('downloads', f"{info['id']}.webm")
+                song_name = info['title']  # Use YouTube video title as the song name
 
         elif uploaded_file:
-            filename = uploaded_file.name  
+            filename = uploaded_file.name
+            song_name = os.path.splitext(uploaded_file.name)[0]  # Use file name as song name
 
         if not filename or not os.path.exists(filename):
             return None, None  # No file, no output
 
-    
-        mp3_filename = "downloads/Ringtone.mp3"
-        audio = AudioSegment.from_file(filename)
-        audio.export(mp3_filename, format="mp3")
-
+        # Convert to MP3
+        mp3_filename = f"downloads/{song_name}.mp3"
+        if not os.path.exists(mp3_filename):
+            audio = AudioSegment.from_file(filename)
+            audio.export(mp3_filename, format="mp3")
         
-        ringtone_filename_m4r = "downloads/Apple.m4r"
-        ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000]
-        ringtone_audio.export(ringtone_filename_m4r, format="mp4")
+        # Create iPhone ringtone (Apple M4R format)
+        ringtone_filename_m4r = f"downloads/{song_name}.m4r"
+        if not os.path.exists(ringtone_filename_m4r):
+            ringtone_audio = AudioSegment.from_file(mp3_filename)[:20000]  # Clip to 20 seconds for ringtone
+            ringtone_audio.export(ringtone_filename_m4r, format="mp4")
+
+        # Save ringtones to Worker and Redis
+        save_ringtone_to_worker(f"{song_name}.mp3", mp3_filename)
+        save_ringtone_to_worker(f"{song_name}.m4r", ringtone_filename_m4r)
+
+        return mp3_filename, ringtone_filename_m4r  # Returning the file paths for download
 
-        return mp3_filename, ringtone_filename_m4r  
-        
     except Exception as e:
         print(f"Error: {e}")
         return None, None
@@ -47,7 +102,7 @@ def process_youtube_url(url, uploaded_file):
 with gr.Blocks() as interface:
     gr.HTML("""
         <h1>Python YouTube Ringtones</h1>
-        <p>Insert a URL to make ringtones or Upload an MP3 to convert</p>
+        <p>Insert a URL to create ringtones or Upload an MP3 to convert.</p>
     """)
     
     with gr.Row():
@@ -61,7 +116,8 @@ with gr.Blocks() as interface:
     process_button = gr.Button("Create Ringtones")
     process_button.click(process_youtube_url, inputs=[youtube_url, mp3_upload], outputs=[mp3_download, iphone_ringtone])
 
-interface.launch()
+interface.launch(share=True)
+