Update app.py

app.py

@@ -303,9 +303,9 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
         if len(comments) < 8:
             gr.Warning("Fewer than 8 comments may lead to errors.")
 
-        #keys =  list(comments.keys())
-        #random.shuffle(keys)
-        #comments = dict([(key, comments[key]) for key in keys])
+        keys =  list(comments.keys())
+        random.shuffle(keys)
+        comments = dict([(key, comments[key]) for key in keys])
 
         prompt = """I will provide a set of artworks along with accompanying comments from a person. Analyze these artworks and the comments on them and identify artistic features such as present or mentioned colors, style, composition, mood, medium, texture, brushwork, lighting, shadow effects, perspective, and other noteworthy elements.
     Your task is to extract the artistic features the person likes and dislikes based on both the artworks' features and the person's comments. Focus solely on artistic aspects and refrain from considering subject matter.
@@ -346,8 +346,6 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
         generated_ids = vpe_model.generate(**inputs, max_new_tokens=2000, repetition_penalty=0.99, do_sample=False)
         generated_texts = processor.batch_decode(generated_ids, skip_special_tokens=True)[0]
 
-        print(generated_texts)
-
         if re.match(r"(.|\n)*Assistant: Liked Art Features: (.|\n)*Disliked Art Features: (.|\n)*", generated_texts):
             positive_vp, negative_vp = re.search('.* \nAssistant: Liked Art Features: (.*)\nDisliked Art Features: (.*)', generated_texts).groups()
             positive_vp = positive_vp.split(", ")
@@ -531,7 +529,8 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
         generator = torch.Generator().manual_seed(seed)
 
         if prompt is None:
-            promtpt = ""
+            gr.Warning("Prompt is empty: a random image will be generated.")
+            prompt = ""
 
         image = pipe(prompt=prompt, 
                     num_inference_steps=num_inference_steps, 
@@ -634,7 +633,7 @@ with gr.Blocks(css=css, title="ViPer Demo", theme=gr.themes.Base()) as demo:
             """
             )
 
-            gr.Markdown("Please write your comments on the images below, explaining why you like or dislike each one from an artistic perspective. Focus on images that evoke **strong reactions**, whether positive or negative, and skip those that don't affect you much.\nMore **detailed** comments will help us provide more personalized results. We recommend commenting on **at least 8** images. If none of the provided images interest you, you can **upload** your own images.")
+            gr.Markdown("Please write your comments on the images below, explaining why you like or dislike each one from an artistic perspective. Comment on a mix of images: some that you like and some that you dislike. Focus on images that evoke strong reactions, whether positive or negative, and skip those that don't affect you much. More detailed comments will help us provide more personalized results. We recommend commenting on at least 8 images. If none of the provided images interest you, you can upload your own images.")
 
 
             with gr.Accordion("Examples of Effective Comments", open=False):