Update app.py

app.py

@@ -44,7 +44,6 @@ def login():
             return redirect(url_for('index'))
         return render_template('login.html', error='Nesprávné heslo')
     return render_template('login.html')
-
 @app.route('/logout')
 def logout():
     session.pop('logged_in', None)
@@ -120,15 +119,20 @@ def translate():
             # Handle YouTube URL
             url = data['url'].strip()
             if 'youtube.com' in url or 'youtu.be' in url:
-                try:
-                    media_file, thumbnail_url = download_youtube_video(url, project_path ) #PROXY_URL, PROXY_USER, PROXY_PASS
-                    # Copy downloaded file to video.mp4
-                    final_path = os.path.join(project_path, "video.mp4")
-                    shutil.copy(media_file, final_path)
-                    media_file = final_path
-                    print("YouTube video downloaded to:", media_file)
-                except Exception as e:
-                    return jsonify({'success': False, 'error': f'YouTube download failed: {str(e)}'})
+                # Check if we already have this video downloaded
+                if 'video_path' in session and os.path.exists(session['video_path']):
+                    media_file = session['video_path']
+                    print("Using already downloaded video:", media_file)
+                else:
+                    try:
+                        media_file, thumbnail_url = download_youtube_video(url, project_path ) #PROXY_URL, PROXY_USER, PROXY_PASS
+                        # Copy downloaded file to video.mp4
+                        final_path = os.path.join(project_path, "video.mp4")
+                        shutil.copy(media_file, final_path)
+                        media_file = final_path
+                        print("YouTube video downloaded to:", media_file)
+                    except Exception as e:
+                        return jsonify({'success': False, 'error': f'YouTube download failed: {str(e)}'})
             else:
                 return jsonify({'success': False, 'error': 'Invalid YouTube URL'})
 
@@ -333,6 +337,16 @@ def download_youtube_video(url, project_path, proxy=None, proxy_user=None, proxy
         print(f"Status code: {response.status_code}")
         print(f"Response headers: {dict(response.headers)}")
         
+        # First check if the response is JSON (error message)
+        content_type = response.headers.get('Content-Type', '').lower()
+        if 'application/json' in content_type:
+            try:
+                error_data = response.json()
+                error_message = error_data.get('error', 'Unknown API error')
+                raise Exception(f"API returned error: {error_message}")
+            except json.JSONDecodeError:
+                pass  # Not JSON, continue with file download
+        
         if response.status_code != 200:
             error_message = f"API request failed with status code {response.status_code}"
             try: