Chandima Prabhath commited on
Commit
f0fa946
·
1 Parent(s): 0e53ecc

Refactor environment variable loading: move dotenv loading to main execution block and set PORT directly

Browse files
Files changed (2) hide show
  1. FLUX.py +3 -3
  2. app.py +1 -1
FLUX.py CHANGED
@@ -5,9 +5,6 @@ import os
5
  import re
6
  import json
7
  from PIL import Image, UnidentifiedImageError
8
- from dotenv import load_dotenv
9
-
10
- load_dotenv()
11
 
12
  # Load the ImgBB API key from the environment variables.
13
  IMGBB_API_KEY = os.getenv("IMGBB_API_KEY")
@@ -130,6 +127,9 @@ def generate_image(prompt, request_id, current_request_id, image_dir, attempt=0)
130
  return image, image_path, returned_prompt, image_url
131
 
132
  if __name__ == "__main__":
 
 
 
133
  # Example usage
134
  prompt = "Beach party, anime style, vibrant colors"
135
  request_id = 1
 
5
  import re
6
  import json
7
  from PIL import Image, UnidentifiedImageError
 
 
 
8
 
9
  # Load the ImgBB API key from the environment variables.
10
  IMGBB_API_KEY = os.getenv("IMGBB_API_KEY")
 
127
  return image, image_path, returned_prompt, image_url
128
 
129
  if __name__ == "__main__":
130
+ from dotenv import load_dotenv
131
+
132
+ load_dotenv()
133
  # Example usage
134
  prompt = "Beach party, anime style, vibrant colors"
135
  request_id = 1
app.py CHANGED
@@ -18,7 +18,7 @@ GREEN_API_MEDIA_URL = os.getenv("GREEN_API_MEDIA_URL", "https://api.green-api.co
18
  GREEN_API_TOKEN = os.getenv("GREEN_API_TOKEN")
19
  GREEN_API_ID_INSTANCE = os.getenv("GREEN_API_ID_INSTANCE")
20
  WEBHOOK_AUTH_TOKEN = os.getenv("WEBHOOK_AUTH_TOKEN")
21
- PORT = int(os.getenv("PORT", 7860))
22
  image_dir = "/tmp/images"
23
  audio_dir = "/tmp/audio"
24
 
 
18
  GREEN_API_TOKEN = os.getenv("GREEN_API_TOKEN")
19
  GREEN_API_ID_INSTANCE = os.getenv("GREEN_API_ID_INSTANCE")
20
  WEBHOOK_AUTH_TOKEN = os.getenv("WEBHOOK_AUTH_TOKEN")
21
+ PORT = 7860
22
  image_dir = "/tmp/images"
23
  audio_dir = "/tmp/audio"
24