Update app.py

app.py

@@ -500,8 +500,8 @@ def stream_logs():
     def generate():
         while True:
             try:
-                # Get log message from queue, timeout after 1 second
-                log_message = log_queue.get()
+                # Get log message from queue, no timeout here, handling is done in the generator
+                log_message = log_queue.get(timeout=1)  # Adding timeout of 1 second to avoid blocking indefinitely
                 yield f"data: {log_message}\n\n"
             except queue.Empty:
                 # Send a heartbeat to keep the connection alive
@@ -513,8 +513,4 @@ def stream_logs():
     response.headers['Cache-Control'] = 'no-cache'
     response.headers['Connection'] = 'keep-alive'
     response.headers['Access-Control-Allow-Origin'] = '*'
-    return response
-
-if __name__ == '__main__':
-    logger.info("Starting Flask API server...")
-    app.run(host='127.0.0.1', port=7860, debug=True, threaded=True)
+    return response