Update app.py

app.py

@@ -11,7 +11,7 @@ from video import create_text_image
 from video2 import video_com,video_func
 
 app = Flask(__name__)
-SUBSPACE_URL = "https://sreepathi-ravikumar-subprocess1.hf.space/generate_video"
+
 SAVE_DIR = "/app/data/video/"
 os.makedirs(SAVE_DIR, exist_ok=True)
 
@@ -86,8 +86,16 @@ def generate_video():
         
         # Print or use lines as slides
         if len(lines)==1:
+            image_folder="/tmp/images"
+            image_olst=[]
+            create_text_image(lines[0],0,image_olst)
+            image_files = sorted(glob.glob(os.path.join(image_folder, "*.png")))
+            if not image_files:
+                raise ValueError("No images found in folder!")
             video_link = "/app/data/video/clip0.mp4"
             video_func(0,lines)
+            for img in image_files:
+                os.remove(img)
             return send_file(video_link, mimetype='video/mp4')
         else:
             for id in range(len(lines)):