Update app.py

app.py

@@ -25,7 +25,7 @@ DESCRIPTION = os.environ.get("DESCRIPTION")
 DISCORD_WEBHOOK = os.environ.get("DISCORD_WEBHOOK")
 TOKEN = os.environ.get("TOKEN")
 
-
+api = HfApi()
 
 def send_discord(i,o):
     url = DISCORD_WEBHOOK
@@ -116,7 +116,7 @@ def predict(message, history, system_prompt, temperature, max_new_tokens, top_k,
     send_discord(instruction, "".join(outputs))
 
 
-    api = HfApi()
+    
     day=datetime.now().strftime("%Y-%m-%d")
     timestamp=datetime.now().timestamp()
     dd={
@@ -162,25 +162,70 @@ async()=>{
 }
 """
 
+def vote(chatbot, data: gr.LikeData):
+    day=datetime.now().strftime("%Y-%m-%d")
+    timestamp=datetime.now().timestamp()
+    api.upload_file(
+        path_or_fileobj=json.dumps({"history":chatbot, 'index': data.index, 'liked': data.liked}, indent=2, ensure_ascii=False).encode('utf-8'),
+        path_in_repo=f"liked/{day}/{timestamp}.json",
+        repo_id="speakleash/bielik-logs",
+        repo_type="dataset",
+        commit_message=f"L",
+        token=TOKEN,
+        run_as_future=True
+    )
+        
 # Create Gradio interface
-gr.ChatInterface(
-    predict,
-    title=EMOJI + " " + MODEL_NAME,
-    description=DESCRIPTION,
-    examples=[
-        ["Kim jesteś?"],
-        ["Ile to jest 9+2-1?"],
-        ["Napisz mi coś miłego."]
-    ],
-    additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False),
-    additional_inputs=[
-        gr.Textbox("", label="System prompt"),
-        gr.Slider(0, 1, 0.6, label="Temperature"),
-        gr.Slider(128, 4096, 1024, label="Max new tokens"),
-        gr.Slider(1, 80, 40, label="Top K sampling"),
-        gr.Slider(0, 2, 1.1, label="Repetition penalty"),
-        gr.Slider(0, 1, 0.95, label="Top P sampling"),
-    ],
-    theme=gr.themes.Soft(primary_hue=COLOR),
-    js=on_load,
-).queue().launch()
+
+with gr.Blocks() as demo:
+    chatbot = gr.Chatbot(label="Chatbot", likeable=True, render=False)
+    chatbot.like(vote, [chatbot], None)
+    gr.ChatInterface(
+        predict,
+        chatbot=chatbot,
+        title=EMOJI + " " + MODEL_NAME,
+        description=DESCRIPTION,
+        examples=[
+            ["Kim jesteś?"],
+            ["Ile to jest 9+2-1?"],
+            ["Napisz mi coś miłego."]
+        ],
+        additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False, render=False),
+        additional_inputs=[
+            gr.Textbox("", label="System prompt", render=False),
+            gr.Slider(0, 1, 0.6, label="Temperature", render=False),
+            gr.Slider(128, 4096, 1024, label="Max new tokens", render=False),
+            gr.Slider(1, 80, 40, label="Top K sampling", render=False),
+            gr.Slider(0, 2, 1.1, label="Repetition penalty", render=False),
+            gr.Slider(0, 1, 0.95, label="Top P sampling", render=False),
+        ],
+        theme=gr.themes.Soft(primary_hue=COLOR),
+        js=on_load,
+    )
+    
+demo.queue().launch()
+        
+# chatbot = gr.Chatbot(label="Chatbot", likeable=True)
+# chatbot.like(vote, None, None)
+# gr.ChatInterface(
+#     predict,
+#     chatbot=chatbot,
+#     title=EMOJI + " " + MODEL_NAME,
+#     description=DESCRIPTION,
+#     examples=[
+#         ["Kim jesteś?"],
+#         ["Ile to jest 9+2-1?"],
+#         ["Napisz mi coś miłego."]
+#     ],
+#     additional_inputs_accordion=gr.Accordion(label="⚙️ Parameters", open=False),
+#     additional_inputs=[
+#         gr.Textbox("", label="System prompt"),
+#         gr.Slider(0, 1, 0.6, label="Temperature"),
+#         gr.Slider(128, 4096, 1024, label="Max new tokens"),
+#         gr.Slider(1, 80, 40, label="Top K sampling"),
+#         gr.Slider(0, 2, 1.1, label="Repetition penalty"),
+#         gr.Slider(0, 1, 0.95, label="Top P sampling"),
+#     ],
+#     theme=gr.themes.Soft(primary_hue=COLOR),
+#     js=on_load,
+# ).queue().launch()