Update app.py

app.py

@@ -16,7 +16,6 @@ def create_history_messages(history):
 def generate_response(prompt, history, model, temperature, max_tokens, top_p, seed):
     messages = create_history_messages(history)
     messages.append({"role": "user", "content": prompt})
-    print(messages)
 
     if seed == 0:
         seed = random.randint(1, 100000)
@@ -41,21 +40,29 @@ def generate_response(prompt, history, model, temperature, max_tokens, top_p, se
 
     return response
 
-def process_video(video_path, text):
-    video = VideoFileClip(video_path).subclip(0, min(60, VideoFileClip(video_path).duration))
+def process_video(text):
+    video_folder = "videos"
+    video_files = [os.path.join(video_folder, f) for f in os.listdir(video_folder) if f.endswith(('mp4', 'mov', 'avi', 'mkv'))]
+    if not video_files:
+        raise FileNotFoundError("No video files found in the specified directory.")
+
+    selected_video = random.choice(video_files)
+    video = VideoFileClip(selected_video)
+    start_time = random.uniform(0, max(0, video.duration - 60))
+    video = video.subclip(start_time, min(start_time + 60, video.duration))
     video = video.resize(height=1920).crop(x1=video.w // 2 - 540, x2=video.w // 2 + 540)
-    
+
     text_lines = text.split()
     text = "\n".join([" ".join(text_lines[i:i+8]) for i in range(0, len(text_lines), 8)])
     
     text_clip = TextClip(text, fontsize=70, color='white', size=video.size, method='caption')
     text_clip = text_clip.set_position('center').set_duration(video.duration)
-    
+
     final = CompositeVideoClip([video, text_clip])
-    
+
     output_path = "output.mp4"
     final.write_videofile(output_path, codec="libx264")
-    
+
     return output_path
 
 additional_inputs = [
@@ -69,7 +76,7 @@ additional_inputs = [
 with gr.Blocks(theme=gr.themes.Soft(primary_hue="red", secondary_hue="pink")) as demo:
     with gr.Tabs():
         with gr.TabItem("Chat"):
-            gr.ChatInterface(
+            chat_interface = gr.ChatInterface(
                 fn=generate_response,
                 chatbot=gr.Chatbot(show_label=False, show_share_button=False, show_copy_button=True, likeable=True, layout="panel"),
                 additional_inputs=additional_inputs,
@@ -77,14 +84,13 @@ with gr.Blocks(theme=gr.themes.Soft(primary_hue="red", secondary_hue="pink")) as
                 description="Powered by GROQ, MoviePy, and other tools.",
             )
         with gr.TabItem("Video Processing"):
-            video_input = gr.Video(label="Upload Video")
             text_input = gr.Textbox(lines=5, label="Text (8 words max per line)")
             process_button = gr.Button("Process Video")
             video_output = gr.Video(label="Processed Video")
 
             process_button.click(
                 fn=process_video,
-                inputs=[video_input, text_input],
+                inputs=[text_input],
                 outputs=video_output,
             )