ahmedbrs commited on
Commit
b5a494e
·
1 Parent(s): fb9fd37

update interface

Browse files
Files changed (2) hide show
  1. app.py +18 -7
  2. output.png +0 -0
app.py CHANGED
@@ -243,11 +243,11 @@ tr:hover {
243
  background: #f7f9f8 !important;
244
  }
245
 
246
- #component-17{
247
  justify-content: center !important;
248
  }
249
 
250
- #component-17 > button {
251
  flex: none !important;
252
  background-color : black !important;
253
  font-weight: bold !important;
@@ -346,23 +346,34 @@ with gr.Blocks(js=scripts, css=css, theme='gstaff/xkcd') as demo:
346
  label="Sketch" , canvas_size=(512,512) , sources=['upload'],
347
  interactive=True , layers= False, transforms=[] )
348
  query_selector = 'button[aria-label="Upload button"]'
349
-
350
- with gr.Row():
351
 
352
  # segment_btn.click(fn=run, inputs=[in_image, in_textbox, in_slider], outputs=[out_image])
353
- upload_draw_btn = gr.HTML(f"""
354
  <div id="upload_draw_group" class="svelte-15lo0d8 stretch">
355
  <button class="sm black-text white-bg gray-border border-radius own-shadow svelte-cmf5ev bold" id="upload_btn" onclick="return document.querySelector('.source-wrap button').click()"> Upload a new sketch</button>
356
  <button class="sm black-text white-bg gray-border border-radius own-shadow svelte-cmf5ev bold" id="draw_btn" onclick="return document.querySelector('.controls-wrap button:nth-child(3)').click()"> Draw a new sketch</button>
357
  </div>
358
  """)
359
- in_textbox = gr.Textbox( lines=2, elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,label="Caption your Sketch!", placeholder="Include the categories that you want the AI to segment. \n e.g. 'giraffe, clouds' or 'a boy flying a kite' ")
 
360
 
361
  with gr.Column():
362
  out_image = gr.Image( value=Image.new('RGB', (512, 512), color=(255, 255, 255)),
363
  elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,
364
  type="pil", label="Segmented Sketch" ) #, height=512, width=512)
365
 
 
 
 
 
 
 
 
 
 
 
 
366
  # gr.HTML("<h3 class='black-text'> <span class='black-text underline'>Confidence:</span> Adjust AI agent confidence in guessing categories </div>")
367
  in_slider = gr.Slider(elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,
368
  info="Adjust AI agent confidence in guessing categories",
@@ -372,7 +383,7 @@ with gr.Blocks(js=scripts, css=css, theme='gstaff/xkcd') as demo:
372
  with gr.Row():
373
  segment_btn = gr.Button( 'Segment it !' , elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" , 'bold' , 'mt-mb-1' ] , size="sm")
374
  segment_btn.click(fn=run, inputs=[in_canvas_image , in_textbox , in_slider ], outputs=[out_image])
375
- gallery_label = gr.HTML("<h3 class='black-text'> <span class='black-text underline'>Gallery:</span> <span style='color: grey;'>you can click on any of the example sketches below</span> </div>")
376
 
377
  gallery= gr.HTML(f"""
378
  <div>
 
243
  background: #f7f9f8 !important;
244
  }
245
 
246
+ #component-19{
247
  justify-content: center !important;
248
  }
249
 
250
+ #component-19 > button {
251
  flex: none !important;
252
  background-color : black !important;
253
  font-weight: bold !important;
 
346
  label="Sketch" , canvas_size=(512,512) , sources=['upload'],
347
  interactive=True , layers= False, transforms=[] )
348
  query_selector = 'button[aria-label="Upload button"]'
349
+ # with gr.Row():
 
350
 
351
  # segment_btn.click(fn=run, inputs=[in_image, in_textbox, in_slider], outputs=[out_image])
352
+ upload_draw_btn = gr.HTML(f"""
353
  <div id="upload_draw_group" class="svelte-15lo0d8 stretch">
354
  <button class="sm black-text white-bg gray-border border-radius own-shadow svelte-cmf5ev bold" id="upload_btn" onclick="return document.querySelector('.source-wrap button').click()"> Upload a new sketch</button>
355
  <button class="sm black-text white-bg gray-border border-radius own-shadow svelte-cmf5ev bold" id="draw_btn" onclick="return document.querySelector('.controls-wrap button:nth-child(3)').click()"> Draw a new sketch</button>
356
  </div>
357
  """)
358
+
359
+ # in_textbox = gr.Textbox( lines=2, elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,label="Caption your Sketch!", placeholder="Include the categories that you want the AI to segment. \n e.g. 'giraffe, clouds' or 'a boy flying a kite' ")
360
 
361
  with gr.Column():
362
  out_image = gr.Image( value=Image.new('RGB', (512, 512), color=(255, 255, 255)),
363
  elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,
364
  type="pil", label="Segmented Sketch" ) #, height=512, width=512)
365
 
366
+ # # gr.HTML("<h3 class='black-text'> <span class='black-text underline'>Confidence:</span> Adjust AI agent confidence in guessing categories </div>")
367
+ # in_slider = gr.Slider(elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,
368
+ # info="Adjust AI agent confidence in guessing categories",
369
+ # label="Confidence:",
370
+ # value=0.5 , interactive=True, step=0.05, minimum=0, maximum=1)
371
+
372
+ with gr.Row():
373
+ with gr.Column():
374
+ in_textbox = gr.Textbox( lines=2, elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,label="Caption your Sketch!", placeholder="Include the categories that you want the AI to segment. \n e.g. 'giraffe, clouds' or 'a boy flying a kite' ")
375
+
376
+ with gr.Column():
377
  # gr.HTML("<h3 class='black-text'> <span class='black-text underline'>Confidence:</span> Adjust AI agent confidence in guessing categories </div>")
378
  in_slider = gr.Slider(elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" ] ,
379
  info="Adjust AI agent confidence in guessing categories",
 
383
  with gr.Row():
384
  segment_btn = gr.Button( 'Segment it !' , elem_classes=["white-bg", "gray-border" , "border-radius" ,"own-shadow" , 'bold' , 'mt-mb-1' ] , size="sm")
385
  segment_btn.click(fn=run, inputs=[in_canvas_image , in_textbox , in_slider ], outputs=[out_image])
386
+ gallery_label = gr.HTML("<h3 class='black-text'> <span class='black-text underline'>Gallery:</span> <span style='color: grey;'>you can click on any of the example sketches below to start segmenting them (or even drawing over them)</span> </div>")
387
 
388
  gallery= gr.HTML(f"""
389
  <div>
output.png CHANGED