serhatderya commited on
Commit
3d9bffc
1 Parent(s): 124a3b5

Update app_scribble.py

Browse files
Files changed (1) hide show
  1. app_scribble.py +21 -5
app_scribble.py CHANGED
@@ -6,17 +6,28 @@ from settings import (DEFAULT_IMAGE_RESOLUTION, DEFAULT_NUM_IMAGES,
6
  MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
7
  from utils import randomize_seed_fn
8
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
9
 
10
  def create_demo(process):
11
  with gr.Blocks() as demo:
12
  with gr.Row():
13
  with gr.Column():
14
  image = gr.Image()
15
- with gr.Row():
16
- with gr.Column():
17
- prompt = gr.Textbox(label='Prompt')
18
- with gr.Column():
19
- prompt_btn = gr.Button("Button")
20
  preprocessor_name = gr.Radio(
21
  label='Preprocessor',
22
  choices=['HED', 'PidiNet', 'None'],
@@ -96,6 +107,11 @@ def create_demo(process):
96
  outputs=result,
97
  api_name=False,
98
  )
 
 
 
 
 
99
  run_button.click(
100
  fn=randomize_seed_fn,
101
  inputs=[seed, randomize_seed],
 
6
  MAX_IMAGE_RESOLUTION, MAX_NUM_IMAGES, MAX_SEED)
7
  from utils import randomize_seed_fn
8
 
9
+ def magic_prompt(prompt):
10
+ openai.api_key = "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"
11
+
12
+ completion = openai.ChatCompletion.create(
13
+ model="gpt-3.5-turbo",
14
+ messages=[
15
+ {"role": "system", "content": "You are a helpful assistant."},
16
+ {"role": "user", "content": "Hello!"}
17
+ ]
18
+ )
19
+
20
+ mag_prompt = str(completion.choices[0].message.content)
21
+ return mag_prompt
22
+
23
 
24
  def create_demo(process):
25
  with gr.Blocks() as demo:
26
  with gr.Row():
27
  with gr.Column():
28
  image = gr.Image()
29
+ prompt = gr.Textbox(label='Prompt')
30
+ mag_prompt_btn = gr.Button("✨Magic Prompt")
 
 
 
31
  preprocessor_name = gr.Radio(
32
  label='Preprocessor',
33
  choices=['HED', 'PidiNet', 'None'],
 
107
  outputs=result,
108
  api_name=False,
109
  )
110
+ mag_prompt_btn.click(
111
+ fn=magic_prompt,
112
+ inputs=prompt,
113
+ outputs=prompt,
114
+ )
115
  run_button.click(
116
  fn=randomize_seed_fn,
117
  inputs=[seed, randomize_seed],