Update app.py

app.py

@@ -57,44 +57,94 @@ 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
-            
-        # ... [your existing line processing code] ...
+        image_folder = "/tmp/images"
+        #line=prompts.splitlines()
+        #asyncio.run(main(line))
+        raw_lines = prompts.splitlines(keepends=False)
+        lines = []
         
-        if True:
-            # Generate the video
-            try:
-                video_func(0, prompt, 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
-            
-
-            # Clean up old files after successful generation
+        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))
+        
+        # Print or use lines as slides
+        if len(lines)==1:
             try:
-                if os.path.exists(audio_link):
-                    os.remove(audio_link)
-            except Exception as cleanup_error:
-                print(f"Cleanup error: {cleanup_error}")
-            
+                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!")
+            video_link1 = "/app/data/video/clip0.mp4"
+            video_link=video_func(0,lines,lang)
+            #for img in image_files:
+                #os.remove(img)
             return send_file(video_link, mimetype='video/mp4')
             
         else:
-            return jsonify({"error": "Multiple lines not yet implemented"}), 501
-            
+            print("learn deeper")
     except Exception as e:
         traceback.print_exc()
         return jsonify({"error": str(e)}), 500