More changes

app.py

@@ -1,8 +1,15 @@
+import logging
+import os
+
 import gradio as gr
+from dotenv import load_dotenv
+
 from utils import add_result
-#import dotenv
 
-HF_TOKEN = ''
+logger = logging.getLogger(__name__)
+logger.setLevel(logging.DEBUG)
+
+load_dotenv()
 
 
 def submit_result(user_answer):
@@ -24,58 +31,69 @@ def get_user_prompt():
         ],
     }
 
-hf_writer = gr.HuggingFaceDatasetSaver(hf_token=HF_TOKEN, dataset_name='maker-faire-bot', private=True)
-csv_writer = gr.CSVLogger()
+
+hf_writer = gr.HuggingFaceDatasetSaver(
+    hf_token=os.environ["HF_TOKEN"], dataset_name="acmc/maker-faire-bot", private=True
+)
+csv_writer = gr.CSVLogger(simplify_file_data=True)
 
 theme = gr.themes.Default(primary_hue="cyan", secondary_hue="fuchsia")
 
 with gr.Blocks(theme=theme) as demo:
     with gr.Row() as header:
         gr.Image(
-            "maker-faire-logo.png",
+            "maker-faire-logo.webp",
             show_download_button=False,
             show_label=False,
             show_share_button=False,
             container=False,
-            #height=100,
-            scale=0.2
+            # height=100,
+            scale=0.2,
         )
         gr.Markdown(
             """
             # Maker Faire Bot
             """,
-            visible=False
+            visible=False,
         )
 
-    # user_prompt = gr.State(get_user_prompt())
-    user_prompt = get_user_prompt()
+    user_prompt = gr.State(get_user_prompt())
 
     gr.Markdown("""# Think about these objects...""")
-    gr.Markdown("""We want to teach the Maker Faire Bot some creativity. Help us get ideas on what you'd build!""")
+    gr.Markdown(
+        """We want to teach the Maker Faire Bot some creativity. Help us get ideas on what you'd build!"""
+    )
     with gr.Row(variant="panel") as row:
-        for i in range(len(user_prompt["images"])):
-            print(i)
+        for i in range(len(user_prompt.value["images"])):
             with gr.Column(variant="default") as col:
                 gr.Image(
-                    user_prompt["images"][i],
-                    label=user_prompt["labels"][i],
+                    user_prompt.value["images"][i],
+                    label=user_prompt.value["labels"][i],
                     interactive=False,
                     show_download_button=False,
                     show_share_button=False,
                 )
-                # gr.Text(user_prompt['labels'][i])
 
-    user_answer = gr.Textbox(
+    user_answer_object = gr.Textbox(
         autofocus=True,
         placeholder="(example): An electronic guitar",
         label="What would you build?",
     )
-    user_answer = gr.TextArea(
-        autofocus=True, label="How would you build it?", placeholder="""I'd use the camera to detect when the user touches the strings and make a sound using the loudspeakers when that happens."""
+    user_answer_explanation = gr.TextArea(
+        autofocus=True,
+        label="How would you build it?",
+        placeholder="""I'd use the camera to detect when the user touches the strings and make a sound using the loudspeakers when that happens.""",
     )
 
+    csv_writer.setup(components=[user_prompt, user_answer_object, user_answer_explanation], flagging_dir="flagged_data_csv")
+    hf_writer.setup(components=[user_prompt, user_answer_object, user_answer_explanation], flagging_dir="flagged_data_hf")
+
     submit_btn = gr.Button("Submit", variant="primary")
-    #submit_btn.click()
+    def log_results(prompt, object, explanation):
+        csv_writer.flag([prompt, object, explanation])
+        hf_writer.flag([prompt, object, explanation])
+
+    submit_btn.click(log_results, inputs=[user_prompt, user_answer_object, user_answer_explanation], preprocess=False)
 
     gr.Markdown(
         """