from flask import Flask, request, jsonify, send_file
from moviepy.editor import ColorClip,ImageClip, concatenate_videoclips,VideoFileClip
import traceback
import uuid
from concurrent.futures import ThreadPoolExecutor, as_completed
import glob
import os
import requests
import asyncio
from image_fetcher import main
from video import create_text_image
from video2 import video_com,video_func

app = Flask(__name__)

SAVE_DIR = "/app/data/video/"
os.makedirs(SAVE_DIR, exist_ok=True)

def send_task(id,url,lines):
    payload = {
        "id": id,
        "lines": lines  
    }

    try:
        response = requests.post(url, json=payload)
        if response.status_code == 200:
            video_path = f"/app/data/video/clip{id}.mp4"
            with open(video_path, "wb") as f:
                f.write(response.content)
            print(video_path)
        else:
            raise Exception(f"[{url}] HTTP {response.status_code}: {response.text}")
    except Exception as e:
        print(f"Error for ID {id}: {e}")
        return None


@app.route("/")
def home():
    return "Flask Video Generator is Running"



@app.route("/generate", methods=["POST"])
def generate_video():
    try:
        data = request.get_json()
        prompt = data.get("duration", '').strip()
        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))
        
        # 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:
            space_b_url = ['https://sreepathi-ravikumar-subprocess1.hf.space/generate', 'https://sreepathi-ravikumar-subprocess2.hf.space/generate', 'https://sreepathi-ravikumar-subprocess3.hf.space/generate', 'https://sreepathi-ravikumar-subprocess4.hf.space/generate', 'https://sreepathi-ravikumar-subprocess5.hf.space/generate', 'https://sreepathi-ravikumar-subprocess6.hf.space/generate', 'https://sreepathi-ravikumar-subprocess7.hf.space/generate', 'https://sreepathi-ravikumar-subprocess8.hf.space/generate', 'https://sreepathi-ravikumar-subprocess9.hf.space/generate', 'https://sreepathi-ravikumar-subprocess10.hf.space/generate']

            with ThreadPoolExecutor(max_workers=len(lines)) as executor:
                for id in range(len(lines)):
                    executor.submit(send_task,id,space_b_url[id],lines)

            video_path = video_com(lines)
            #for img in image_files:
                #os.remove(img)
            return send_file(video_path, mimetype='video/mp4')
    except Exception as e:
        traceback.print_exc()
        return jsonify({"error": str(e)}), 500

if __name__ == "__main__":
    app.run(host="0.0.0.0", port=7860)

    # Example call (remove or change in your actual app)