Update app.py

app.py

@@ -48,6 +48,21 @@ OPENROUTER_MODEL = "mistralai/mistral-small-3.1-24b-instruct:free"
 OUTPUT_VIDEO_FILENAME = "final_video.mp4"
 USER_AGENT = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
 
+
+
+# Additional global variables needed for the Gradio interface
+selected_voice = 'af_heart'  # Default voice
+voice_speed = 0.9  # Default voice speed
+font_size = 45  # Default font size
+video_clip_probability = 0.25  # Default probability for video clips
+bg_music_volume = 0.08  # Default background music volume
+fps = 30  # Default FPS
+preset = "veryfast"  # Default preset
+TARGET_RESOLUTION = None
+CAPTION_COLOR = None
+TEMP_FOLDER = None
+
+
 # ---------------- Helper Functions ---------------- #
 # (Your existing helper functions remain unchanged: generate_script, parse_script,
 # search_pexels_videos, search_pexels_images, search_google_images, download_image,
@@ -434,7 +449,7 @@ def generate_media(prompt, user_image=None, current_index=0, total_segments=1):
         else:
             print(f"Google Images search failed for prompt: {prompt}")
 
-    if random.random() < 0.50:
+    if random.random() < video_clip_probability:
         video_file = os.path.join(TEMP_FOLDER, f"{safe_prompt}_video.mp4")
         video_url = search_pexels_videos(prompt, PEXELS_API_KEY)
         if video_url:
@@ -758,12 +773,37 @@ def fix_imagemagick_policy():
         print(f"Error fixing policies: {e}")
         return False
 
-# ---------------- Main Function with Gradio Integration ---------------- #
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+
+# ---------------- Main Video Generation Function ---------------- #
 def generate_video(user_input, resolution, caption_option):
     """Generate a video based on user input via Gradio."""
     global TARGET_RESOLUTION, CAPTION_COLOR, TEMP_FOLDER
-    import shutil
-
+    
     # Set resolution
     if resolution == "Full":
         TARGET_RESOLUTION = (1920, 1080)
@@ -852,21 +892,7 @@ def generate_video(user_input, resolution, caption_option):
 
     return OUTPUT_VIDEO_FILENAME
 
-
-
-
-
-
-
-
-
-
-
-
-import gradio as gr
-import shutil
-import os
-
+# ---------------- Gradio Interface ---------------- #
 VOICE_CHOICES = {
     'Emma (Female)': 'af_heart',
     'Bella (Female)': 'af_bella',
@@ -898,28 +924,28 @@ VOICE_CHOICES = {
     'Daniel 🇬🇧 (Male)': 'bm_daniel'
 }
 
-
-def handle_music_upload(music_file):
+def generate_video_with_options(user_input, resolution, caption_option, music_file, voice, vclip_prob, bg_vol, video_fps, video_preset, v_speed, caption_size):
+    global selected_voice, voice_speed, font_size, video_clip_probability, bg_music_volume, fps, preset
+    
+    # Update global variables with user selections
+    selected_voice = VOICE_CHOICES[voice]
+    voice_speed = v_speed
+    font_size = caption_size
+    video_clip_probability = vclip_prob / 100  # Convert from percentage to decimal
+    bg_music_volume = bg_vol
+    fps = video_fps
+    preset = video_preset
+    
+    # Handle music upload
     if music_file is not None:
         target_path = "music.mp3"
-        shutil.copy(music_file, target_path)
+        shutil.copy(music_file.name, target_path)
         print(f"Uploaded music saved as: {target_path}")
-
-def generate_video_with_options(user_input, resolution, caption_option, music_file, voice, video_clip_probability, bg_music_volume, fps, preset, voice_speed, font_size):
-    handle_music_upload(music_file)
-
-    selected_voice = VOICE_CHOICES[voice]
-    print(f"Selected Voice: {selected_voice}")
-    print(f"Video Clip Probability: {video_clip_probability}%")
-    print(f"Music Volume: {bg_music_volume}")
-    print(f"FPS: {fps}")
-    print(f"Preset: {preset}")
-    print(f"Voice Speed: {voice_speed}")
-    print(f"Caption Font Size: {font_size}")
-
-    # Pass these to your video generator as needed
+    
+    # Generate the video
     return generate_video(user_input, resolution, caption_option)
 
+# Create the Gradio interface
 iface = gr.Interface(
     fn=generate_video_with_options,
     inputs=[
@@ -931,14 +957,16 @@ iface = gr.Interface(
         gr.Slider(0, 100, value=25, step=1, label="Video Clip Usage Probability (%)"),
         gr.Slider(0.0, 1.0, value=0.08, step=0.01, label="Background Music Volume"),
         gr.Slider(10, 60, value=30, step=1, label="Video FPS"),
-        gr.Dropdown(choices=["ultrafast", "superfast", "veryfast", "faster", "fast", "medium", "slow"], value="veryfast", label="Export Preset"),
+        gr.Dropdown(choices=["ultrafast", "superfast", "veryfast", "faster", "fast", "medium", "slow"], 
+                   value="veryfast", label="Export Preset"),
         gr.Slider(0.5, 1.5, value=0.9, step=0.05, label="Voice Speed"),
         gr.Slider(20, 100, value=45, step=1, label="Caption Font Size")
     ],
     outputs=gr.Video(label="Generated Video"),
     title="AI Documentary Video Generator",
-    description="Upload music, choose voice, control video clips, volume, FPS, speed, captions."
+    description="Create short documentary videos with AI. Upload music, choose voice, and customize settings."
 )
 
-iface.launch(share=True)
-
+# Launch the interface
+if __name__ == "__main__":
+    iface.launch(share=True)