updated code; added theme

app.py

@@ -24,24 +24,26 @@ def get_IF_op(prompt, neg_prompt):
   file_list = os.listdir(folder_path)
   file_list = [os.path.join(folder_path, f) for f in file_list if f != 'captions.json']
   print(f"^^file list is: {file_list}")
-  return file_list   
+  return file_list    
 
-def get_pickscores(prompt, file_list):
+def get_pickscores(prompt, image_tmps):
   print("inside get_pickscores")
   #Get the predictons
-  probabilities1 = client_pick.predict(prompt, file_list[0], file_list[1], fn_index=0)
-  probabilities2 = client_pick.predict(prompt, file_list[2], file_list[3], fn_index=0)
+  probabilities1 = client_pick.predict(prompt, image_tmps[0], image_tmps[1], fn_index=0)
+  probabilities2 = client_pick.predict(prompt, image_tmps[2], image_tmps[3], fn_index=0)
   probabilities_all = list(probabilities1) + list(probabilities2)
   max_score = max(probabilities_all)
   max_score_index = probabilities_all.index(max_score)
-  best_match_image = file_list[max_score_index]
+  best_match_image = image_tmps[max_score_index]
   return best_match_image
 
+
 def get_upscale_op(prompt, gallery_if):
   print("inside get_upscale_op")
   print(f"^^gallery_if is: {gallery_if}")
+  image_tmps = [val['name'] for val in gallery_if]
   # get pickscores
-  best_match_image = get_pickscores(prompt, gallery_if)
+  best_match_image = get_pickscores(prompt, image_tmps)
   # let's get the best pick! 
   low_res_img = Image.open(best_match_image).convert("RGB")
   low_res_img = low_res_img.resize((128, 128))
@@ -50,18 +52,20 @@ def get_upscale_op(prompt, gallery_if):
   #upscaled_image.save("upsampled.png")
   return upscaled_image
 
-with gr.Blocks() as demo:
+with gr.Blocks(theme=gr.themes.Monochrome()) as demo:
   with gr.Row():
-    with gr.Column():
+    with gr.Column(scale=4):
       prompt = gr.Textbox(label='Prompt')
       neg_prompt = gr.Textbox(label='Negative Prompt')
-    b1 = gr.Button('Generate')
-  with gr.Row():
-      gallery_if = gr.Gallery(label='IF Space outputs', ).style(preview=True, rows=[1], object_fit="contain", height="400px")
+    with gr.Column(scale=1):
+      b1 = gr.Button('Generate').style(full_width=True)
+  gallery_if = gr.Gallery(label='IF Space outputs', ).style(columns=4, object_fit="contain", preview=True)
   b2 = gr.Button("Get the best generation using Pick-A-Pic")
   image_picakapic = gr.Image(label="PickAPic Evaluated Output") 
 
   b1.click(get_IF_op,[prompt, neg_prompt], gallery_if)
+  prompt.submit(get_IF_op,[prompt, neg_prompt], gallery_if)
   b2.click(get_upscale_op,[prompt, gallery_if], image_picakapic)
 
-demo.launch(debug=True)
+
+demo.launch()