Update app.py

app.py

@@ -9,13 +9,22 @@ import random
 from PIL import Image
 import io
 import base64
+from dotenv import load_dotenv
+
+# Load environment variables from the .env file
+load_dotenv()
 
 # Initialize Flask app
 app = Flask(__name__)
 
-# Set server and websocket addresses
-server_address = "127.0.0.1:8188"
-ws_address = "ws://127.0.0.1:8188/ws"
+# Set server and websocket addresses from environment variables
+server_address = os.getenv("SERVER_ADDRESS")
+ws_address = os.getenv("WS_ADDRESS")
+
+# Example print statements to verify the addresses are loaded
+print(f"Server Address: {server_address}")
+print(f"WebSocket Address: {ws_address}")
+
 client_id = str(uuid.uuid4())
 
 def make_request(url, data=None, headers=None):