Update app.py

app.py

@@ -1,11 +1,10 @@
-from flask import Flask, request, jsonify, Response
+from flask import Flask, request, jsonify, Response, stream_with_context
 import requests
 import json
 import time
 import random
 import logging
 import sys
-import asyncio
 
 app = Flask(__name__)
 
@@ -155,18 +154,16 @@ def handle_request():
         image_data = {'data': [{'url': image_url}]}
         
         if stream:
-            return stream_response(request, unique_id, image_data, prompt, enhanced_prompt, "1024x1024", current_timestamp, model, system_fingerprint)
+            return stream_response(unique_id, image_data, prompt, enhanced_prompt, "1024x1024", current_timestamp, model, system_fingerprint)
         else:
             return non_stream_response(unique_id, image_data, prompt, enhanced_prompt, "1024x1024", current_timestamp, model, system_fingerprint)
     except Exception as e:
         logger.error(f"Unexpected error in handle_request: {str(e)}")
         return jsonify({"error": f"Internal Server Error: {str(e)}"}), 500
 
-def stream_response(request, unique_id, image_data, original_prompt, translated_prompt, size, created, model, system_fingerprint):
+def stream_response(unique_id, image_data, original_prompt, translated_prompt, size, created, model, system_fingerprint):
     logger.debug("Starting stream response")
-    response = Response(stream_with_context(generate_stream(unique_id, image_data, original_prompt, translated_prompt, size, created, model, system_fingerprint)), content_type='text/event-stream')
-    logger.debug("Stream response completed")
-    return response
+    return Response(stream_with_context(generate_stream(unique_id, image_data, original_prompt, translated_prompt, size, created, model, system_fingerprint)), content_type='text/event-stream')
 
 def generate_stream(unique_id, image_data, original_prompt, translated_prompt, size, created, model, system_fingerprint):
     chunks = [