Update app.py

app.py

@@ -437,7 +437,7 @@ def stitch_media(
   video_file, video_url,
   audio_file, audio_url,
   subtitle_file, subtitle_url,
-  book_id,
+  book_id, book_cover_file, book_cover_url,
   enable_highlight,
   highlight_color,
   font_size,
@@ -499,17 +499,24 @@ def stitch_media(
 
       subtitle_escaped = subtitle_to_use.replace('\\', '/').replace(':', '\\:')
 
-      has_book_cover = book_id and book_id.strip()
+      has_book_cover = (book_id and book_id.strip()) or book_cover_file or book_cover_url
       timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
       output_path = os.path.join(temp_dir, f"final_{timestamp}.mp4")
 
       fade_color_hex = "#dacfc3"
 
       if has_book_cover:
-          status_msg += f"\n📚 Downloading book cover (ID: {book_id})...\n"
           try:
-              book_cover_path = download_book_cover(book_id.strip(), temp_dir)
-              status_msg += "✅ Book cover downloaded\n"
+              if book_cover_file:
+                  status_msg += f"\n📚 Using uploaded book cover...\n"
+                  book_cover_path = book_cover_file
+              elif book_cover_url:
+                  status_msg += f"\n📚 Downloading book cover from URL...\n"
+                  book_cover_path = download_file_from_url(book_cover_url.strip(), temp_dir, 'book_cover_from_url.jpg')
+              else:
+                  status_msg += f"\n📚 Downloading book cover (ID: {book_id})...\n"
+                  book_cover_path = download_book_cover(book_id.strip(), temp_dir)
+              status_msg += "✅ Book cover ready\n"
 
               fade_starts_at = audio_duration * VIDEO_CONFIG['fade_start_percent']
               fade_ends_at = audio_duration * VIDEO_CONFIG['fade_end_percent']
@@ -693,6 +700,8 @@ async def stitch_upload(
     audio_url: Optional[str] = Form(None),
     subtitle_url: Optional[str] = Form(None),
     book_id: Optional[str] = Form(None),
+    book_cover_file: Optional[UploadFile] = File(None),
+    book_cover_url: Optional[str] = Form(None),
     enable_highlight: bool = Form(True),
     highlight_color: str = Form('yellow'),
     font_size: int = Form(12),
@@ -707,6 +716,8 @@ async def stitch_upload(
         'audio_url': audio_url,
         'subtitle_url': subtitle_url,
         'book_id': book_id,
+        'book_cover_file': None,
+        'book_cover_url': book_cover_url,
         'enable_highlight': enable_highlight,
         'highlight_color': highlight_color,
         'font_size': font_size,
@@ -720,6 +731,8 @@ async def stitch_upload(
             payload['audio_file'] = _save_upload_to_temp(audio_file, temp_dir)
         if subtitle_file is not None:
             payload['subtitle_file'] = _save_upload_to_temp(subtitle_file, temp_dir)
+        if book_cover_file is not None:
+            payload['book_cover_file'] = _save_upload_to_temp(book_cover_file, temp_dir)
 
         start_time = time.time()
 
@@ -731,7 +744,8 @@ async def stitch_upload(
                 payload.get('video_file'), payload.get('video_url'),
                 payload.get('audio_file'), payload.get('audio_url'),
                 payload.get('subtitle_file'), payload.get('subtitle_url'),
-                payload.get('book_id'), payload.get('enable_highlight', True),
+                payload.get('book_id'), payload.get('book_cover_file'), payload.get('book_cover_url'),
+                payload.get('enable_highlight', True),
                 payload.get('highlight_color', 'yellow'), payload.get('font_size', 12),
                 payload.get('crf_quality', 23)
             )