Update app.py

app.py

@@ -141,60 +141,53 @@ import os
 
 # ... आपके अन्य सभी इम्पोर्ट और एंडपॉइंट ...
 
+from flask import Flask, request, jsonify
+from piped_api import PipedClient
+import time
+
+app = Flask(__name__)
+PIPED_CLIENT = PipedClient()
+
 @app.route('/get-stream', methods=['POST'])
 def get_stream():
-    """YouTube से ऑडियो स्ट्रीम URL निकालकर लौटाएं (रीट्राई के साथ)"""
+    start_time = time.time()
     data = request.json
     query = data.get('query')
     if not query:
         return jsonify({"error": "Query is required"}), 400
 
-    # yt-dlp की सेटिंग्स (एक बार डिफाइन करें)
-    ydl_opts = {
-        'format': 'bestaudio/best[height<=720]',
-        'quiet': True,
-        'no_warnings': True,
-        'extract_flat': False,
-        'force_ipv4': True,  # 👈 IPv6 के झंझट से बचने के लिए
-        'socket_timeout': 30,  # 👈 टाइमआउट बढ़ाया
-        'http_headers': {
-            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/121.0.0.0 Safari/537.36',
-        },
-        'extractor_args': {'youtube': {'player_client': ['android', 'web']}},
-        'cookiefile': 'cookies.txt' if os.path.exists('cookies.txt') else None,
-    }
-
-    # रीट्राई लॉजिक (3 बार कोशिश करें)
-    max_retries = 3
-    for attempt in range(max_retries):
-        try:
-            with yt_dlp.YoutubeDL(ydl_opts) as ydl:
-                info = ydl.extract_info(f"ytsearch:{query}", download=False)
-                if not info or 'entries' not in info or not info['entries']:
-                    return jsonify({"error": "No video found"}), 404
-
-                video = info['entries'][0]
-                stream_url = video.get('url')
-                if not stream_url:
-                    return jsonify({"error": "Could not extract stream URL"}), 500
-
-                return jsonify({
-                    "title": video.get('title'),
-                    "webpage_url": video.get('webpage_url'),
-                    "duration": video.get('duration'),
-                    "stream_url": stream_url,
-                    "requester": data.get('requester', 'Unknown')
-                })
-
-        except Exception as e:
-            print(f"❌ Attempt {attempt+1} failed for '{query}': {e}")
-            if attempt < max_retries - 1:
-                time.sleep(2)  # थोड़ा रुककर फिर कोशिश करें
-                continue
-            else:
-                # सारी कोशिशें फेल होने पर एरर भेजें
-                return jsonify({"error": f"All retries failed: {str(e)}"}), 500
+    try:
+        # Piped API से केवल 1 रिजल्ट खोजें (तेज़ी के लिए)
+        search_results = PIPED_CLIENT.search(query, limit=1)
+        if not search_results:
+            return jsonify({"error": "No video found"}), 404
+
+        first_result = search_results[0]
+        video_id = first_result['url'].split('=')[-1]
+
+        # वीडियो की डिटेल लें
+        video_detail = PIPED_CLIENT.get_video(video_id)
+        if not video_detail.audio_streams:
+            return jsonify({"error": "No audio stream found"}), 500
+
+        # सबसे अच्छी क्वालिटी की ऑडियो स्ट्रीम चुनें
+        audio_stream = video_detail.audio_streams[0]
         
+        processing_time = time.time() - start_time
+        print(f"✅ Stream URL generated in {processing_time:.2f} seconds")
+
+        return jsonify({
+            "title": video_detail.title,
+            "webpage_url": f"https://youtube.com/watch?v={video_id}",
+            "duration": video_detail.duration,
+            "stream_url": audio_stream.url,
+            "requester": data.get('requester', 'Unknown')
+        })
+
+    except Exception as e:
+        print(f"❌ Error in /get-stream: {e}")
+        return jsonify({"error": str(e)}), 500
+
 # ---------- TTS ----------
 @app.route('/tts', methods=['POST'])
 def tts_api():