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 |
|