video size choice and more debugging metrics

app.py

@@ -23,6 +23,9 @@ def estimate_chunk_durations(text, words_per_second=2.5, min_sec=5, max_sec=10):
                 current_duration = 0
     if current_chunk:
         chunks.append(" ".join(current_chunk))
+    
+    total_time = sum([min(max(len(chunk.split()) / words_per_second, min_sec), max_sec) for chunk in chunks])
+    print(f"Total estimated time for video: {total_time:.2f} seconds")
     return chunks
 
 def generate_speech(text):
@@ -31,47 +34,51 @@ def generate_speech(text):
     tts.tts_to_file(text=text, file_path=wav_path)
     return wav_path
 
-def generate_images(chunks):
+def generate_images(chunks, image_size=(640, 480)):
     pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4")
     pipe.to("cuda" if torch.cuda.is_available() else "cpu")
     image_paths = []
     for i, chunk in enumerate(chunks):
+        print(f"Generating image for chunk {i + 1} of {len(chunks)}: {chunk[:50]}...")  # Printing part of the chunk
         image = pipe(chunk).images[0]
+        image = image.resize(image_size)
         img_path = f"image_{i}.png"
         image.save(img_path)
         image_paths.append(img_path)
     return image_paths
 
-def create_video(images, durations, speech_path):
-    clips = [mp.ImageClip(img).set_duration(dur) for img, dur in zip(images, durations)]
-    black_start = mp.ColorClip((512, 512), color=(0,0,0), duration=1)
-    black_end = mp.ColorClip((512, 512), color=(0,0,0), duration=2)
+def create_video(images, durations, speech_path, image_size=(640, 480)):
+    clips = [mp.ImageClip(img).set_duration(dur).resize(image_size) for img, dur in zip(images, durations)]
+    black_start = mp.ColorClip(image_size, color=(0,0,0), duration=1)
+    black_end = mp.ColorClip(image_size, color=(0,0,0), duration=2)
     video = mp.concatenate_videoclips([black_start] + clips + [black_end])
     audio = mp.AudioFileClip(speech_path)
     final_video = video.set_audio(audio)
     final_video.write_videofile("output.mp4", fps=24)
     return "output.mp4"
 
-def process_text(text):
+def process_text(text, image_size):
     chunks = estimate_chunk_durations(text)
     speech_path = generate_speech(text)
-    image_paths = generate_images(chunks)
+    image_paths = generate_images(chunks, image_size)
     durations = [min(10, max(5, len(chunk.split()) / 2.5)) for chunk in chunks]
-    video_path = create_video(image_paths, durations, speech_path)
+    video_path = create_video(image_paths, durations, speech_path, image_size)
     return video_path
 
 with gr.Blocks() as demo:
     gr.Markdown("# Text-to-Video Generator using AI 🎥")
     text_input = gr.Textbox(label="Enter your text")
     file_input = gr.File(label="Or upload a .txt file")
+    image_size_input = gr.Radio(choices=["640x480", "800x600", "1024x768"], label="Select Image Size", value="640x480")
     process_btn = gr.Button("Generate Video")
     output_video = gr.Video()
     
-    def handle_request(text, file):
+    def handle_request(text, file, image_size):
         if file is not None:
             text = open(file.name, "r").read()
-        return process_text(text)
+        image_size_dict = {"640x480": (640, 480), "800x600": (800, 600), "1024x768": (1024, 768)}
+        return process_text(text, image_size_dict[image_size])
     
-    process_btn.click(handle_request, inputs=[text_input, file_input], outputs=output_video)
+    process_btn.click(handle_request, inputs=[text_input, file_input, image_size_input], outputs=output_video)
 
 demo.launch()

sample_text.txt

@@ -0,0 +1,15 @@
+Ancient Egypt: A Glimpse into a Lost Civilization
+
+Ancient Egypt, one of the most fascinating and enduring civilizations in history, flourished along the Nile River for over 3,000 years. Its civilization developed around 3100 BCE and lasted until Alexander the Great conquered Egypt in 332 BCE. Egypt is famous for its monumental achievements in architecture, art, and culture, many of which have had a lasting influence on the world.
+
+The civilization was known for its unique religious beliefs, with gods and goddesses playing a central role in daily life. The ancient Egyptians believed in a complex system of deities, each representing various aspects of life and nature. One of the most famous gods was Ra, the sun god, believed to be the ruler of all gods. The Egyptians also believed in the afterlife, where the soul would continue to exist if one lived a righteous life.
+
+The Egyptians’ architectural achievements are also awe-inspiring, with the Great Pyramids of Giza standing as one of the Seven Wonders of the Ancient World. These massive structures were built as tombs for the pharaohs, the god-kings who ruled Egypt. The pyramids symbolize the Egyptians’ belief in the afterlife, where the pharaohs would need a place to rest for eternity.
+
+Hieroglyphics, the writing system of Ancient Egypt, is another remarkable achievement. This system of pictorial symbols allowed the Egyptians to record important events, religious texts, and administrative details. Many of the texts written in hieroglyphics have been preserved on tombs, monuments, and papyri, providing us with a window into the daily lives and beliefs of the ancient Egyptians.
+
+Another major aspect of Ancient Egypt was its economy, which was primarily based on agriculture. The Nile River’s predictable flooding provided fertile soil for farming, allowing the Egyptians to grow crops like wheat, barley, and flax. The abundance of food helped Egypt maintain a stable and prosperous society.
+
+Throughout its long history, Ancient Egypt experienced periods of political stability and turmoil. It was divided into the Old, Middle, and New Kingdoms, with each period characterized by different rulers and societal changes. Some of the most famous pharaohs, including Tutankhamun, Ramses II, and Cleopatra, lived during these different periods.
+
+The ancient Egyptians left a profound legacy that still influences modern culture. From the grandeur of the pyramids to the enduring mystery of the Sphinx, Egypt's monuments continue to awe and inspire people around the world.

sample_text_short.txt

@@ -0,0 +1,3 @@
+Ancient Egypt: A Glimpse into a Lost Civilization
+
+Ancient Egypt, one of the most fascinating and enduring civilizations in history, flourished along the Nile River for over 3,000 years. Its civilization developed around 3100 BCE and lasted until Alexander the Great conquered Egypt in 332 BCE. Egypt is famous for its monumental achievements in architecture, art, and culture, many of which have had a lasting influence on the world.