Update app.py

app.py

@@ -1,45 +1,106 @@
-from flask import Flask, request, jsonify, Response,redirect
+from flask import Flask, request, jsonify, send_file
 from flask_cors import CORS
-import time
-import random
-from text2generation import generate_long_answer  # Your own function
+from moviepy.editor import ImageClip, concatenate_videoclips
+import traceback
+import uuid
+import glob
+import os
+import asyncio
+
+from text2generation import generate_long_answer  # Your function
+from image_fetcher import main  # Your async function (optional)
+from video import create_text_image  # Your function
 
 app = Flask(__name__)
 CORS(app)
-temp_data={}
+
+# Store temporary data
+temp_data = {}
 
 @app.route('/health')
 def health_check():
     return jsonify({"status": "ready"}), 200
 
-# Standard answer route
 @app.route('/ask', methods=['POST'])
 def ask_question():
     try:
-        global temp_data
         data = request.get_json()
         question = data.get('question', '').strip()
 
         if not question:
             return jsonify({"error": "Question is required"}), 400
 
+        # Step 1: Generate Answer
         answer = generate_long_answer(question)
-        temp_data['answered']=answer
-        return jsonify({"question": question, "answer": answer})
+        temp_data['answer'] = answer
+
+        # Step 2: Start Video Generation
+        video_path = generate_video_from_text(answer)
+        temp_data['video_path'] = video_path
+
+        # Step 3: Respond with JSON
+        return jsonify({
+            "question": question,
+            "answer": answer,
+            "video_url": f"/download_video/{os.path.basename(video_path)}"
+        })
+
+    except Exception as e:
+        traceback.print_exc()
+        return jsonify({"error": str(e)}), 500
 
+@app.route('/download_video/<filename>')
+def download_video(filename):
+    try:
+        path = f"/tmp/{filename}"
+        return send_file(path, mimetype='video/mp4')
     except Exception as e:
         return jsonify({"error": str(e)}), 500
 
-@app.route('/get_video')
-def get_video():
-    video_lst=[
-        "https://videos.pexels.com/video-files/3571264/3571264-sd_640_360_30fps.mp4",
-        "https://videos.pexels.com/video-files/3052446/3052446-hd_1920_1080_30fps.mp4",
-        "https://videos.pexels.com/video-files/6394054/6394054-sd_960_480_24fps.mp4",
-        
-    ]
-    return redirect(random.choice(video_lst))
+def generate_video_from_text(prompts):
+    image_folder = "/tmp/images"
+    os.makedirs(image_folder, exist_ok=True)
+
+    raw_lines = prompts.splitlines(keepends=False)
+    lines = []
+
+    i = 0
+    while i < len(raw_lines):
+        line = raw_lines[i].strip()
+        if line.endswith('?') or line.endswith(';'):
+            block = line
+            i += 1
+            paragraph_lines = []
+            count = 0
+            while i < len(raw_lines) and not (raw_lines[i].strip().endswith('?') or raw_lines[i].strip().endswith(';')) and count < 5:
+                paragraph_lines.append(raw_lines[i])
+                i += 1
+                count += 1
+            if paragraph_lines:
+                block += '\n' + '\n'.join(paragraph_lines)
+            lines.append(block)
+        else:
+            block = '\n'.join(raw_lines[i:i+5])
+            lines.append(block)
+            i += 5
+
+    image_list = []
+    for idx, block in enumerate(lines):
+        create_text_image(block, idx, image_list)
+
+    image_files = sorted(glob.glob(os.path.join(image_folder, "*.png")))
+    if not image_files:
+        raise ValueError("No images found!")
+
+    clips = [ImageClip(img).set_duration(5) for img in image_files]
+    video = concatenate_videoclips(clips, method="compose")
+    video_path = f"/tmp/video_{uuid.uuid4().hex}.mp4"
+    video.write_videofile(video_path, fps=24)
+
+    for img in image_files:
+        os.remove(img)
 
+    return video_path
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860)