Update app.py

app.py

@@ -1,99 +1,73 @@
 import gradio as gr
 import yt_dlp
-import os
 import tempfile
+import os
 import shutil
-import uuid  # برای نام منحصربه‌فرد فایل‌ها
 
-def convert_to_mp3(url):
+def convert_to_mp3(video_url):
     """
-    دانلود صوت از لینک ویدیو و تبدیل به MP3.
-    خروجی: path فایل MP3 و پیام (شامل راهنمایی لینک).
+    دانلود ویدیو/صوت از لینک و تبدیل به MP3.
+    فایل موقت MP3 ایجاد می‌شود و لینک دانلود برمی‌گردد.
     """
-    if not url:
-        return None, "لطفاً لینک ویدیو را وارد کنید."
-    
-    # ایجاد پوشه موقت
-    temp_dir = tempfile.mkdtemp()
-    unique_id = str(uuid.uuid4())  # نام منحصربه‌فرد برای جلوگیری از تداخل
-    output_template = f'{unique_id}_%(title)s.%(ext)s'
-    output_path = os.path.join(temp_dir, output_template)
+    if not video_url:
+        return None, "لینک ویدیو را وارد کنید."
     
+    # تنظیمات yt-dlp برای دانلود صوت و تبدیل مستقیم به MP3
     ydl_opts = {
         'format': 'bestaudio/best',  # بهترین کیفیت صوت
-        'extractaudio': True,        # استخراج صوت
-        'audioformat': 'mp3',        # تبدیل به MP3
-        'outtmpl': output_path,      # نام فایل
-        'postprocessors': [{         # پردازش با FFmpeg
-            'key': 'FFmpegExtractAudio',
+        'postprocessors': [{
+            'key': 'FFmpegExtractAudio',  # استخراج صوت با FFmpeg
             'preferredcodec': 'mp3',
-            'preferredquality': '192',  # 192kbps
+            'preferredquality': '192',  # کیفیت 192kbps (قابل تنظیم)
         }],
-        'noplaylist': True,          # فقط ویدیو اصلی
+        'outtmpl': 'temp.%(ext)s',  # نام فایل موقت
+        'quiet': True,  # کاهش لاگ‌ها
     }
     
     try:
         with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-            info = ydl.extract_info(url, download=False)
-            title = info.get('title', 'audio').replace('/', '_').replace('\\', '_')  # پاکسازی نام
-            ydl.download([url])
+            ydl.download([video_url])
         
-        # پیدا کردن فایل MP3
-        mp3_files = [f for f in os.listdir(temp_dir) if f.startswith(unique_id) and f.endswith('.mp3')]
-        if not mp3_files:
-            shutil.rmtree(temp_dir, ignore_errors=True)
-            return None, "خطا در دانلود یا تبدیل فایل."
+        # پیدا کردن فایل MP3 (پس از postprocessor)
+        mp3_file = None
+        for file in os.listdir('.'):
+            if file.startswith('temp.') and file.endswith('.mp3'):
+                mp3_file = file
+                break
         
-        mp3_filename = mp3_files[0]
-        mp3_path = os.path.join(temp_dir, mp3_filename)
+        if not mp3_file:
+            return None, "خطا در دانلود یا تبدیل. مطمئن شوید لینک معتبر است."
         
-        # پیام Markdown با راهنمایی لینک (برای UI و API)
-        message = f"""
-        ### ✅ تبدیل موفق!
-        - **فایل**: `{mp3_filename}`
+        # کپی به فایل موقت برای Gradio (برای امنیت و تمیزکاری)
+        temp_mp3 = tempfile.NamedTemporaryFile(suffix='.mp3', delete=False)
+        temp_mp3.close()  # بستن برای نوشتن
+        shutil.copy2(mp3_file, temp_mp3.name)
         
-        #### 🔗 لینک دانلود MP3
-        - **در UI Gradio**: از دکمه دانلود در کامپوننت File استفاده کنید (URL موقت تولید می‌شود).
-        - **در API**: خروجی `data[0]` یک URL مستقیم به MP3 است (مثل `https://your-space.hf.space/file={mp3_filename}`).
+        # تمیز کردن فایل اصلی
+        os.remove(mp3_file)
         
-        **نکته مهم**: لینک‌ها موقت هستند (در Spaces، ۱-۲ ساعت اعتبار). برای API، به `/api/predict` POST کنید با `data: [{{"text": "لینک ویدیو"}}]`.
-        """
-        
-        return mp3_path, message
+        return temp_mp3.name, "فایل MP3 با موفقیت ایجاد شد. دانلود کنید."
     
     except Exception as e:
-        shutil.rmtree(temp_dir, ignore_errors=True)
-        return None, f"### ❌ خطا!\n{str(e)}\n\n**نکته**: مطمئن شوید لینک معتبر است (مثل یوتیوب). yt-dlp از بسیاری سایت‌ها پشتیبانی می‌کند."
+        return None, f"خطا: {str(e)}"
 
-# اینترفیس Gradio
-demo = gr.Interface(
+# رابط Gradio
+iface = gr.Interface(
     fn=convert_to_mp3,
     inputs=gr.Textbox(
         label="لینک ویدیو (مثل یوتیوب)",
-        placeholder="https://www.youtube.com/watch?v=...",
-        lines=1
+        placeholder="https://www.youtube.com/watch?v=..."
     ),
     outputs=[
-        gr.File(
-            label="فایل MP3 (دانلود)",
-            visible=True  # دکمه دانلود نمایش داده می‌شود
-        ),
-        gr.Markdown(label="پیام و لینک")  # بدون lines – تصحیح شده
+        gr.File(label="دانلود MP3"),
+        gr.Textbox(label="پیام وضعیت")
     ],
-    title="تبدیل ویدیو به MP3 با لینک دانلود",
-    description="""لینک ویدیو را وارد کنید. خروجی شامل فایل MP3 و لینک دانلود موقت است. 
-    در API، URL MP3 مستقیماً برمی‌گردد. از yt-dlp و FFmpeg استفاده می‌کند.
-
-    **استفاده از API**:
-    - POST به `https://your-space.hf.space/api/predict`
-    - Body: `{"data": [{"text": "https://youtube-link..."}]}` 
-    - Response: `{"data": ["path/to/mp3.mp3", "message"]}` – `data[0]` URL پروکسی‌شده است.
-    """,
+    title="تبدیل ویدیو به MP3",
+    description="لینک ویدیو را وارد کنید تا صوت آن به MP3 تبدیل شود.",
     examples=[
-        ["https://www.youtube.com/watch?v=dQw4w9WgXcQ"],
-    ],
-    theme=gr.themes.Soft()
+        ["https://www.youtube.com/watch?v=dQw4w9WgXcQ"]
+    ]
 )
 
 if __name__ == "__main__":
-    demo.launch(share=True)
+    iface.launch()