Update app.py

app.py

@@ -1,62 +1,48 @@
 import streamlit as st
-from moviepy.editor import ImageClip, concatenate_videoclips, TextClip, CompositeVideoClip
+from moviepy.editor import *
 from PIL import Image
-from io import BytesIO
-import os
 import tempfile
+import os
 
-st.set_page_config(page_title="Image to Video Creator", layout="wide")
-st.title("🖼️➡️🎞️ Image to Video Generator with Context")
+st.set_page_config(page_title="Image to Video", layout="centered")
+st.title("🖼️➡️🎞️ Convert Images to Video with Captions")
 
-uploaded_files = st.file_uploader("Upload Images", type=["jpg", "png", "jpeg"], accept_multiple_files=True)
+uploaded_files = st.file_uploader("Upload Images", type=["jpg", "jpeg", "png"], accept_multiple_files=True)
 
-context_list = []
+duration = st.slider("Duration per image (sec)", 2, 10, 4)
 
 if uploaded_files:
-    st.subheader("📝 Enter Context or Caption for Each Image")
-
-    for i, img_file in enumerate(uploaded_files):
-        st.image(img_file, width=200, caption=f"Image {i+1}")
-        context = st.text_area(f"Context for Image {i+1}", key=f"context_{i}")
-        context_list.append(context)
-
-    duration = st.slider("Duration per image (seconds)", 2, 10, 4)
-
-    if st.button("Create Video"):
-        with st.spinner("Generating video..."):
+    captions = []
+    st.subheader("Add a caption for each image:")
+    for i, image_file in enumerate(uploaded_files):
+        st.image(image_file, width=150)
+        cap = st.text_input(f"Caption for image {i+1}", key=f"caption_{i}")
+        captions.append(cap)
+
+    if st.button("Generate Video"):
+        with st.spinner("Processing..."):
             temp_dir = tempfile.mkdtemp()
             clips = []
 
-            for i, img_file in enumerate(uploaded_files):
-                # Save image temporarily
-                img = Image.open(img_file).convert("RGB")
-                img_path = os.path.join(temp_dir, f"img_{i}.png")
+            for i, file in enumerate(uploaded_files):
+                img = Image.open(file).convert("RGB")
+                img_path = os.path.join(temp_dir, f"img_{i}.jpg")
                 img.save(img_path)
 
-                # Create ImageClip
-                image_clip = ImageClip(img_path).set_duration(duration)
+                clip = ImageClip(img_path).set_duration(duration)
 
-                # Add context text if any
-                if context_list[i]:
-                    txt_clip = TextClip(context_list[i], fontsize=24, color='white', bg_color='black', size=image_clip.size)
-                    txt_clip = txt_clip.set_duration(duration).set_position(("center", "bottom"))
-                    final_clip = CompositeVideoClip([image_clip, txt_clip])
-                else:
-                    final_clip = image_clip
+                if captions[i]:
+                    txt = TextClip(captions[i], fontsize=24, color='white', bg_color='black', size=clip.size)
+                    txt = txt.set_duration(duration).set_position(("center", "bottom"))
+                    clip = CompositeVideoClip([clip, txt])
 
-                clips.append(final_clip)
+                clips.append(clip)
 
             final_video = concatenate_videoclips(clips, method="compose")
-            output_path = os.path.join(temp_dir, "final_video.mp4")
-            final_video.write_videofile(output_path, fps=24)
+            video_path = os.path.join(temp_dir, "output.mp4")
+            final_video.write_videofile(video_path, fps=24)
 
-            with open(output_path, "rb") as file:
-                video_bytes = file.read()
+            with open(video_path, "rb") as f:
+                video_bytes = f.read()
                 st.video(video_bytes)
-                st.download_button("Download Video", data=video_bytes, file_name="image_video.mp4", mime="video/mp4")
-
-        st.success("✅ Video created successfully!")
-
-else:
-    st.info("Upload some images to begin.")
-
+                st.download_button("Download Video", video_bytes, file_name="video.mp4", mime="video/mp4")