Added: Generate image to video

.gitignore

@@ -0,0 +1 @@
+/static

app.py

@@ -63,7 +63,7 @@ def save_base64_image(b64_string):
         with open(image_path, 'wb') as f:
             f.write(image_data)
 
-        print(f"Image saved at: {image_path}")
+        print(f"Image saved at: {image_path}", flush=True)
 
         # Return the path and URL of the saved image
         image_url = f"https://gosign-de-comfyui-api.hf.space/{image_path}"
@@ -235,6 +235,7 @@ def get_video_data(filename, subfolder, token):
 
     # Build the URL with encoded query parameters
     url = f"{server_address}/view?{urllib.parse.urlencode(url_values)}"
+
     print(f"Requesting URL: {url}", flush=True)
 
     # Prepare the request with authorization token
@@ -264,6 +265,15 @@ def get_video_data(filename, subfolder, token):
 def image_to_video():
     data = request.json
 
+    time.sleep(900)
+    # Prepare the response with the video URL
+    response_data = {
+            'video_url': 'https://gosign-de-comfyui-stg.hf.space/view?filename=CogVideoX-I2V_00010.mp4',
+            'message': 'Video generated successfully'
+        }
+
+    return jsonify(response_data), 200
+
     # Extract and validate token
     token = request.headers.get('Authorization')
     if not token or not token.startswith("Bearer "):
@@ -367,18 +377,30 @@ def image_to_video():
             raise ValueError('Failed to generate video')
 
         # Save the video locally
-        local_video_path = f"static/generated_video_{uuid.uuid4()}.mp4"
+        local_video_path = f"static/generated_video_{prompt_id}.mp4"
         with open(local_video_path, 'wb') as f:
             f.write(video_data)
+        # Construct the public URL for the video
+        video_url = f"https://gosign-de-comfyui-api.hf.space/{local_video_path}"
+
+        # Prepare the response with the video URL
+        response_data = {
+            'video_url': video_url,
+            'message': 'Video generated successfully'
+        }
+
+        return jsonify(response_data), 200
 
         # Send the video as an HTTP response
-        return send_file(
+        response = send_file(
             io.BytesIO(video_data),
             mimetype='video/mp4',
             as_attachment=True,
             download_name='generated_video.mp4'
         )
 
+        return response
+
     except Exception as e:
         return jsonify({'error': str(e)}), 500
 
@@ -386,9 +408,7 @@ def image_to_video():
         # Always delete the image if it was saved
         if image_path and os.path.exists(image_path):
             os.remove(image_path)
-            print(f"Deleted temporary image: {image_path}")
-
-    # return response
+            print(f"Deleted temporary image: {image_path}", flush=True)
 
 if __name__ == '__main__':
     app.run(host='0.0.0.0', port=7860, debug=True)  # Removed 'debug=True'