DjPapzin commited on
Commit
bb82692
·
1 Parent(s): e58a9b7

Improve environment variable handling for API keys

Browse files
Files changed (1) hide show
  1. app.py +11 -0
app.py CHANGED
@@ -1,7 +1,18 @@
1
  import os
2
  import gradio as gr
 
3
  from src.app import DryingApp
4
 
 
 
 
 
 
 
 
 
 
 
5
  # Initialize the Drying Assistant application
6
  app = DryingApp()
7
 
 
1
  import os
2
  import gradio as gr
3
+ from dotenv import load_dotenv
4
  from src.app import DryingApp
5
 
6
+ # Load environment variables
7
+ load_dotenv()
8
+
9
+ # Check if API keys are available
10
+ if not os.getenv("OPENROUTER_API_KEY"):
11
+ print("Warning: OPENROUTER_API_KEY environment variable not set")
12
+
13
+ if not os.getenv("STABILITY_API_KEY"):
14
+ print("Warning: STABILITY_API_KEY environment variable not set")
15
+
16
  # Initialize the Drying Assistant application
17
  app = DryingApp()
18