Update app.py

app.py

@@ -62,6 +62,7 @@ repo_name = os.environ["LOG_REPO"]
 
 external_log_dir = "./logs"
 LOGDIR = external_log_dir
+VOTEDIR = "./votes"
 
 
 def install_gradio_4_35_0():
@@ -87,6 +88,36 @@ def get_conv_log_filename():
     name = os.path.join(LOGDIR, f"{t.year}-{t.month:02d}-{t.day:02d}-user_conv.json")
     return name
 
+def get_conv_vote_filename():
+    t = datetime.datetime.now()
+    name = os.path.join(VOTEDIR, f"{t.year}-{t.month:02d}-{t.day:02d}-user_vote.json")
+    if not os.path.isfile(name):
+        os.makedirs(os.path.dirname(name), exist_ok=True)
+    return name
+
+def vote_last_response(state, vote_type, model_selector):
+    with open(get_conv_vote_filename(), "a") as fout:
+        data = {
+            "type": vote_type,
+            "model": model_selector,
+            "state": state,
+        }
+        fout.write(json.dumps(data) + "\n")
+    api.upload_file(
+        path_or_fileobj=get_conv_vote_filename(),
+        path_in_repo=get_conv_vote_filename().replace("./votes/", ""),
+        repo_id=repo_name,
+        repo_type="dataset")
+
+
+def upvote_last_response(state):
+    vote_last_response(state, "upvote", "Pangea-7b")
+    return state
+
+def downvote_last_response(state):
+    vote_last_response(state, "downvote", "Pangea-7b")
+    return state
+
 class InferenceDemo(object):
     def __init__(
         self, args, model_path, tokenizer, model, image_processor, context_len
@@ -298,6 +329,14 @@ def bot(history, temperature, top_p, max_output_tokens):
     our_chatbot.conversation.append_message(our_chatbot.conversation.roles[1], None)
     prompt = our_chatbot.conversation.get_prompt()
 
+    if len(images_this_term) == 0:
+        gr.Warning("You should upload an image. Please upload the image and try again.")
+        return history
+    
+    if len(images_this_term) > 1:
+        gr.Warning("Only one image can be uploaded in a conversation. Please reduce the number of images and start a new conversation.")
+        return history
+
     # input_ids = (
     #     tokenizer_image_token(
     #         prompt, our_chatbot.tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt"
@@ -648,6 +687,15 @@ with gr.Blocks(
         fn=clear_history, inputs=[chatbot], outputs=[chatbot], api_name="clear_all"
     )
 
+    upvote_btn.click(
+        fn=upvote_last_response, inputs=[chatbot], outputs=[chatbot], api_name="upvote_last_response"
+    )
+
+    
+    downvote_btn.click(
+        fn=downvote_last_response, inputs=[chatbot], outputs=[chatbot], api_name="upvote_last_response"
+    )
+
 
 demo.queue()