Update app.py

app.py

@@ -1,6 +1,7 @@
 import gradio as gr
-# from gradio_huggingfacehub_search import HuggingfaceHubSearch
 import pandas as pd
+from gradio_huggingfacehub_search import HuggingfaceHubSearch
+import requests
 
 
 def update_table(category):
@@ -25,17 +26,49 @@ def update_table(category):
     df = pd.DataFrame(data)
     return df
 
-def submit_vote(vote):
-    # code
-    return f"Vote '{vote}' submitted successfully!"
+def get_user(profile: gr.OAuthProfile | None) -> str:
+    if profile is None:
+        return ""
+    return f"Hello {profile.username}"
 
-def submit_model(model_id):
-    if not model_id:
+def update_vote_ui(profile: gr.OAuthProfile | None):
+    username = get_user(profile)
+    if username:
+        username_widget = gr.Textbox(value=username, label="Username", interactive=False)
+    else:
+        username_widget = gr.LoginButton()
+    return username_widget
+
+
+def submit_vote(username, category, vote):
+    if not category or not vote:
         return "All fields are required!"
-    
+    if not username:
+        return "You need to log in to submit a vote."
+    if username.startswith("Hello "):
+        username = username[6:]
+
     url = "https://sdk.nexa4ai.com/task"   # Will have a new endpoint
+    data = {
+        "category": category,
+        "model_id": vote,
+        "username": username
+    }
+    
+    response = requests.post(url, json=data)
+    
+    if response.status_code == 200:
+        return f"Vote '{vote}' submitted successfully!"
+    else:
+        return f"Failed to vote: {response.text}"
 
+def submit_model(category, model_id):
+    if not category or not model_id:
+        return "All fields are required!"
+    
+    url = "https://sdk.nexa4ai.com/task"   # Will have a new endpoint
     data = {
+        "category": category,
         "model_id": model_id
     }
     
@@ -47,17 +80,14 @@ def submit_model(model_id):
         return f"Failed to submit request: {response.text}"
 
 
+# with gr.Blocks(auth=gr.HuggingfaceOAuth(optional=True)) as app:
 with gr.Blocks() as app:
     with gr.Tabs():
         with gr.TabItem("Table"):
-            dropdown = gr.Dropdown(
-                choices=[
-                    "Overall", "Biology", "Physics", "Business", "Chemistry",
-                    "Economics", "Philosophy", "History", "Culture", "Computer Science",
-                    "Math", "Health", "Law", "Engineering", "Other"
-                ],
+            category = gr.Dropdown(
+                choices=["Overall", "Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Health", "Law", "Engineering", "Other"],
                 label="Select Category",
-                value="Overall"  
+                value="Overall"
             )
             
             initial_data = update_table("Overall")
@@ -66,24 +96,35 @@ with gr.Blocks() as app:
                 datatype=["number", "str", "number", "str", "str"],
                 value=initial_data,
                 col_count=(5, "fixed"),
-                # height=600
             )
-            dropdown.change(update_table, inputs=dropdown, outputs=table)
+            category.change(update_table, inputs=category, outputs=table)
         
         with gr.TabItem("Vote"):
-            vote = gr.Radio(choices=["Option 1", "Option 2", "Option 3"], label="Choose your option")
+            profile = None
+            if profile:
+                username = get_user(profile)
+                username_text = gr.Textbox(value=username, label="Username", interactive=False)
+            else:
+                login_button = gr.LoginButton()
+
+            category = gr.Dropdown(
+                choices=["Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Health", "Law", "Engineering", "Other"], 
+                label="Select Category"
+            )
+            vote = gr.CheckboxGroup(choices=["Option 1", "Option 2", "Option 3"], label="Choose your options")
             submit_button = gr.Button("Submit Vote")
-            submit_result = gr.Label() 
-            submit_button.click(fn=submit_vote, inputs=vote, outputs=submit_result)
+            submit_result = gr.Label()
+            if profile:
+                submit_button.click(fn=submit_vote, inputs=[username_text, category, vote], outputs=submit_result)
+            else:
+                submit_button.click(fn=lambda: "Please log in to submit your vote.", inputs=[], outputs=submit_result)
+
         
-        # with gr.TabItem("Submit Model"):
-        #     model_id = HuggingfaceHubSearch(
-        #         label="Hub Model ID",
-        #         placeholder="Search for model id on Huggingface",
-        #         search_type="model",
-        #     )
-        #     submit_model_button = gr.Button("Submit Model")
-        #     submit_model_result = gr.Label()
-        #     submit_model_button.click(fn=submit_model, inputs=[model_id], outputs=submit_model_result)
+        with gr.TabItem("Submit Model"):
+            category = gr.Dropdown(choices=["Biology", "Physics", "Business", "Chemistry", "Economics", "Philosophy", "History", "Culture", "Computer Science", "Math", "Health", "Law", "Engineering", "Other"], label="Select Category")
+            model_id = HuggingfaceHubSearch(label="Hub Model ID", placeholder="Search for model id on Huggingface", search_type="model")
+            submit_model_button = gr.Button("Submit Model")
+            submit_model_result = gr.Label()
+            submit_model_button.click(fn=submit_model, inputs=[category, model_id], outputs=submit_model_result)
 
-app.launch()
+app.launch()