Upload 2 files

app.py

@@ -11,6 +11,9 @@ from openai import OpenAI
 import gradio as gr
 import huggingface_hub
 from datasets import load_dataset
+from wordcloud import WordCloud
+import matplotlib.pyplot as plt
+import ast
 
 api_key = os.environ.get("API_TOKEN")
 headers = {
@@ -112,30 +115,27 @@ def simulate(image0, image1):
     avg_preference = sum(preferences) / len(preferences)
     avg_prob = sum(probs) / len(preferences)
 
-    return avg_preference
-
-subtitle = "Upload two images of emails and see which is generally preferred by Petco customers!"
-
-
-from pathlib import Path
+    preference = 0 if avg_preference < 0.5 else 1
+    
+    right_reasons = [reasons[i] for i in len(reasons) if preferences[i] == preference]
+    reasons_list = ast.literal_eval(right_reasons)
+    all_reasons = ','.join(reasons_list)
+    wordcloud = WordCloud(width = 800, height = 400, background_color ='white').generate(all_reasons)
 
-def upload_file(filepath):
-    name = Path(filepath).name
-    return [gr.UploadButton(visible=False), gr.DownloadButton(label=f"{name}", value=filepath, visible=True)]
+    plt.figure(figsize=(10,5))
+    plt.imshow(wordcloud, interpolation='bilinear')
+    plt.axis("off")
+    plt.show()
 
-# with gr.Blocks() as demo:
-#     gr.Markdown("First upload a file and and then you'll be able download it (but only once!)")
-#     with gr.Row():
-#         u = gr.UploadButton("Upload a file", file_count="single")
-#         d = gr.DownloadButton("Download the file", visible=False)
+    return preference, plt
 
-#     u.upload(upload_file, u, [u, d])
-#     d.click(download_file, None, [u, d])
+subtitle = "Upload two images of emails and see which is generally preferred by Petco customers!"
 
 demo = gr.Interface(fn=simulate, 
                     inputs=[gr.UploadButton("Click to Upload Email 0", file_types=["image"], file_count="single"), 
                             gr.UploadButton("Click to Upload Email 1", file_types=["image"], file_count="single")], 
-                    outputs="text",
+                    outputs=["text",
+                            gr.Plot(value=plt)],
                     title="Pairwise Simulation of Petco Email Preference",
                     description=subtitle
                     )

requirements.txt

@@ -2,4 +2,5 @@ openai
 https://gradio-builds.s3.amazonaws.com/75ed61524f55e542414d165ed60a976091c24f77/gradio-3.16.2-py3-none-any.whl
 pandas
 numpy
-datasets
+datasets
+wordcloud