Update app.py

app.py

@@ -57,94 +57,53 @@ def home():
 
 
 
+@app.route("/generate", methods=["POST"])
 @app.route("/generate", methods=["POST"])
 def generate_video():
     try:
         data = request.get_json()
         userprompt = data.get("duration", '').strip()
-        prompt=extract_english_paragraphs(userprompt)
-        lang=extract_native_text(userprompt)
-        prompts=prompt.replace("**","")
+        prompt = extract_english_paragraphs(userprompt)
+        lang = extract_native_text(userprompt)
+        prompts = prompt.replace("**", "")
         print(prompts)
+        
         if prompts == '':
             return jsonify({"error": "prompts be must"}), 400
-        image_folder = "/tmp/images"
-        #line=prompts.splitlines()
-        #asyncio.run(main(line))
-        raw_lines = prompts.splitlines(keepends=False)
-        lines = []
-        
-        i = 0
-        while i < len(raw_lines):
-            line = raw_lines[i].strip()
-        
-            # Check if current line is a heading
-            if line.strip().startswith("#") and (line.endswith('?') or line.endswith(':')):
-                block = line  # Start block with heading
-                i += 1
-        
-                # Accumulate body lines until next heading or 5+ lines
-                paragraph_lines = []
-                while i < len(raw_lines):
-                    next_line = raw_lines[i].strip()
-        
-                    # Stop if next line is a heading
-                    if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
-                        break
-        
-                    paragraph_lines.append(next_line)
-                    i += 1
-        
-                    # If we've gathered enough lines for a slide, break to next
-                    if len(paragraph_lines) >= 5:
-                        break
-        
-                # Combine heading + paragraph
-                if paragraph_lines:
-                    block += '\n' + '\n'.join(paragraph_lines)
-        
-                lines.append(block)
-        
-            else:
-                # Group normal lines (not part of any heading)
-                block_lines = []
-                count = 0
-        
-                while i < len(raw_lines) and count < 5:
-                    next_line = raw_lines[i].strip()
-        
-                    # If this is a heading, break to handle it separately
-                    if next_line.strip().startswith("#") and (next_line.endswith('?') or next_line.endswith(':')):
-                        break
-        
-                    block_lines.append(next_line)
-                    i += 1
-                    count += 1
-        
-                if block_lines:
-                    lines.append('\n'.join(block_lines))
+            
+        # ... [your existing line processing code] ...
         
-        # Print or use lines as slides
-        if len(lines)==1:
-            try:
-                os.remove("/app/data/video/clip0.mp4")
-                os.remove("/app/data/sound/audio0.mp3")
-            except Exception as e:
-                print(e)
-            #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!")
+        if len(lines) == 1:
             video_link = "/app/data/video/clip0.mp4"
-            video_func(0,lines,lang)
-            #for img in image_files:
-                #os.remove(img)
+            audio_link = "/app/data/sound/audio0.mp3"
+            
+            # Generate the video
+            try:
+                video_func(0, lines, lang)
+            except Exception as video_error:
+                print(f"Video generation failed: {video_error}")
+                traceback.print_exc()
+                return jsonify({"error": f"Video generation failed: {str(video_error)}"}), 500
+            
+            # Verify the file was created and has content
+            if not os.path.exists(video_link):
+                return jsonify({"error": "Video file was not created"}), 500
+                
+            if os.path.getsize(video_link) == 0:
+                return jsonify({"error": "Video file is empty"}), 500
+            
+            # Clean up old files after successful generation
+            try:
+                if os.path.exists(audio_link):
+                    os.remove(audio_link)
+            except Exception as cleanup_error:
+                print(f"Cleanup error: {cleanup_error}")
+            
             return send_file(video_link, mimetype='video/mp4')
             
         else:
-            print("learn deeper")
+            return jsonify({"error": "Multiple lines not yet implemented"}), 501
+            
     except Exception as e:
         traceback.print_exc()
         return jsonify({"error": str(e)}), 500