Update app.py

app.py

@@ -100,33 +100,38 @@ def create_video(images, durations, audio_path):
     video.write_videofile(output_path, fps=1, codec='libx264')
     return output_path
 
-def process_pipeline(prompt, progress=gr.Progress(track_tqdm=True)):
+def process_pipeline(prompt, progress=gr.Progress()):
     try:
-        with progress.tqdm(total=6) as pbar:
-            pbar.set_description("Generating Story")
+        with progress:
+            total_steps = 6
+            step = 0
+
+            progress(step / total_steps, desc="Generating Story")
             story = generate_story(prompt)
-            pbar.update(1)
-            
-            pbar.set_description("Splitting Story into Sentences")
+            step += 1
+
+            progress(step / total_steps, desc="Splitting Story into Sentences")
             sentences = split_story_into_sentences(story)
-            pbar.update(1)
-            
-            pbar.set_description("Generating Images for Sentences")
+            step += 1
+
+            progress(step / total_steps, desc="Generating Images for Sentences")
             images = generate_images(sentences)
-            pbar.update(1)
-            
-            pbar.set_description("Generating Audio")
+            step += 1
+
+            progress(step / total_steps, desc="Generating Audio")
             audio_path, total_duration = generate_audio(story)
-            pbar.update(1)
-            
-            pbar.set_description("Computing Durations")
+            step += 1
+
+            progress(step / total_steps, desc="Computing Durations")
             durations = compute_sentence_durations(sentences, total_duration)
-            pbar.update(1)
-            
-            pbar.set_description("Creating Video")
+            step += 1
+
+            progress(step / total_steps, desc="Creating Video")
             video_path = create_video(images, durations, audio_path)
-            pbar.update(1)
-            
+            step += 1
+
+            progress(1.0, desc="Completed")
+
         return video_path
     except Exception as e:
         print(f"Error in process_pipeline: {str(e)}")