Spaces:
Running
Running
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
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 =
|
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 |
|