Upload folder using huggingface_hub

app.py

@@ -30,18 +30,16 @@ use_real_api = False
 
 # Function to generate a paper_id using SHA-512 hash
 def generate_paper_id(paper_content):
-    if isinstance(paper_content, str):
-        paper_content = paper_content.encode('utf-8')
     return hashlib.sha512(paper_content).hexdigest()
 
 # Function to get user IP address
-# def get_user_ip():
-#     try:
-#         return request.remote_addr
-#     except:
-#         return "Unknown"
+def get_user_ip():
+    try:
+        return request.remote_addr
+    except:
+        return "Unknown"
 
-def review_papers(pdf_file, user_ip):
+def review_papers(pdf_file):
     logging.info(f"Received file type: {type(pdf_file)}")
     paper_content = pdf_file.read() if hasattr(pdf_file, 'read') else pdf_file  # Read the content of the uploaded PDF file
     if use_real_api:
@@ -103,8 +101,8 @@ def review_papers(pdf_file, user_ip):
     logging.debug(f"Final formatted reviews: {review_texts}")
     return review_texts[0], review_texts[1], gr.update(visible=True), gr.update(visible=True), model_a, model_b, paper_content
 
-def handle_vote(vote, model_a, model_b, paper_content, user_ip):
-    user_id = user_ip  # Use the passed user IP address as user_id
+def handle_vote(vote, model_a, model_b, paper_content):
+    user_id = get_user_ip()  # Get the user IP address as user_id
     paper_id = generate_paper_id(paper_content)  # Generate paper_id from paper content
     
     # Write the request
@@ -167,17 +165,6 @@ def setup_interface():
     }
     """
     with gr.Blocks(css=css) as demo:
-        gr.HTML("""
-            <input type="hidden" id="user-ip" value="">
-            <script>
-                async function getUserIP() {
-                    const response = await fetch('https://api.ipify.org?format=json');
-                    const data = await response.json();
-                    document.getElementById('user-ip').value = data.ip;
-                }
-                window.onload = getUserIP;
-            </script>
-        """)
         with gr.Tabs():
             with gr.TabItem("Reviewer Arena"):
                 gr.Markdown("## Reviewer Arena")
@@ -201,14 +188,14 @@ def setup_interface():
 
                 model_identity_message = gr.HTML("", visible=False)
 
-                def handle_vote_interface(vote, model_identity_message_a, model_identity_message_b, paper_content, user_ip):
-                    return handle_vote(vote, model_identity_message_a, model_identity_message_b, paper_content, user_ip)
+                def handle_vote_interface(vote, model_identity_message_a, model_identity_message_b, paper_content):
+                    return handle_vote(vote, model_identity_message_a, model_identity_message_b, paper_content)
 
-                submit_button.click(fn=review_papers, inputs=[file_input, gr.Textbox(visible=False, value=lambda: document.getElementById('user-ip').value)],
-                    outputs=[review1, review2, vote, vote_button, model_identity_message, model_identity_message])
+                submit_button.click(fn=review_papers, inputs=[file_input],
+                                    outputs=[review1, review2, vote, vote_button, model_identity_message, model_identity_message])
 
-                vote_button.click(fn=handle_vote_interface, inputs=[vote, model_identity_message, model_identity_message, file_input, gr.Textbox(visible=False, value=lambda: document.getElementById('user-ip').value)],
-                                    outputs=[vote_message, vote, vote_button, another_paper_button])
+                vote_button.click(fn=handle_vote_interface, inputs=[vote, model_identity_message, model_identity_message],
+                                  outputs=[vote_message, vote, vote_button, another_paper_button])
 
                 another_paper_button.click(fn=lambda: None, inputs=None, outputs=None, js="() => { location.reload(); }")
 
@@ -218,7 +205,6 @@ def setup_interface():
                 # Fetch the leaderboard data from the database
                 leaderboard_data = get_leaderboard()
                 print(leaderboard_data)
-                    
                 
                 # Create the leaderboard HTML dynamically
                 leaderboard_html = """
@@ -264,9 +250,7 @@ def setup_interface():
     return demo
 
 
-
-
 if __name__ == "__main__":
     logging.basicConfig(level=logging.INFO)
     demo = setup_interface()
-    demo.launch()
+    demo.launch()