Update app.py

app.py

@@ -64,6 +64,7 @@ repo_name = os.environ["LOG_REPO"]
 
 external_log_dir = "./logs"
 LOGDIR = external_log_dir
+VOTEDIR = "./votes"
 
 
 def install_gradio_4_35_0():
@@ -89,6 +90,37 @@ 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", "PULSE-7B")
+    return state
+
+def downvote_last_response(state):
+    vote_last_response(state, "downvote", "PULSE-7B")
+    return state
+
+
 class InferenceDemo(object):
     def __init__(
         self, args, model_path, tokenizer, model, image_processor, context_len
@@ -212,6 +244,7 @@ def add_message(history, message):
         history.append(((x,), None))
     if message["text"] is not None:
         history.append((message["text"], None))
+        
     return history, gr.MultimodalTextbox(value=None, interactive=False)
 
 
@@ -242,10 +275,11 @@ def bot(history, temperature, top_p, max_output_tokens):
     # for message in history[-i-1:]:
     #     images_this_term.append(message[0][0])
 
+    
     assert len(images_this_term) > 0, "must have an image"
     # image_files = (args.image_file).split(',')
     # image = [load_image(f) for f in images_this_term if f]
-
+        
     all_image_hash = []
     all_image_path = []
     for image_path in images_this_term:
@@ -296,13 +330,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()
 
-    # input_ids = (
-    #     tokenizer_image_token(
-    #         prompt, our_chatbot.tokenizer, IMAGE_TOKEN_INDEX, return_tensors="pt"
-    #     )
-    #     .unsqueeze(0)
-    #     .to(our_chatbot.model.device)
-    # )
+    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"
         ).unsqueeze(0).to(our_chatbot.model.device)
@@ -502,7 +537,17 @@ 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()
     
 if __name__ == "__main__":