Upload app.py

app.py

@@ -7,7 +7,7 @@ DATA_DIR = "submissions"
 os.makedirs(DATA_DIR, exist_ok=True)
 
 # Predefined task types
-TASK_TYPES = ["Classification", "Regression", "Translation", "Custom"]
+TASK_TYPES = ["math", "Open-Ended", "Custom"]
 
 # Function to handle task submission
 def submit_task(task_type, description, yaml_text):
@@ -36,6 +36,13 @@ def get_tasks_by_type(task_type):
                 tasks.append(data)
     return tasks
 
+# Function to dynamically add a new task type
+def add_new_task_type(new_type):
+    if new_type and new_type not in TASK_TYPES:
+        TASK_TYPES.append(new_type)
+        return gr.Dropdown.update(choices=TASK_TYPES), f"Task type '{new_type}' added successfully!"
+    return gr.Dropdown.update(choices=TASK_TYPES), "Task type already exists or invalid input."
+
 # Function to switch tabs
 def switch_tab(tab_name):
     return gr.Tabs.update(visible_tab=tab_name)
@@ -52,9 +59,17 @@ with gr.Blocks() as app:
             task_type_input = gr.Dropdown(
                 label="Task Type",
                 choices=TASK_TYPES,
-                value="Classification",
+                value="math",
+                interactive=True
+            )
+            new_task_input = gr.Textbox(
+                label="Add New Task Type",
+                placeholder="Enter a new task type",
                 interactive=True
             )
+            add_task_button = gr.Button("Add Task Type")
+            add_task_status = gr.Textbox(label="Status", interactive=False)
+
             description_input = gr.Textbox(
                 label="Task Description",
                 placeholder="Provide a brief description of the task.",
@@ -69,6 +84,13 @@ with gr.Blocks() as app:
             go_to_view_tab = gr.Button("Go to View Tasks")
             submission_status = gr.Textbox(label="Status", interactive=False)
 
+            # Handle adding new task type
+            add_task_button.click(
+                add_new_task_type,
+                inputs=[new_task_input],
+                outputs=[task_type_input, add_task_status]
+            )
+
             # Handle task submission
             submit_button.click(
                 submit_task,
@@ -83,14 +105,13 @@ with gr.Blocks() as app:
                 outputs=[tabs]
             )
 
-
         with gr.Tab("View Tasks"):
             gr.Markdown("## View Submitted Tasks")
 
             task_type_filter = gr.Dropdown(
                 label="Filter by Task Type",
                 choices=TASK_TYPES,
-                value="Classification",
+                value="math",
                 interactive=True
             )
             view_button = gr.Button("View Tasks")
@@ -116,5 +137,4 @@ with gr.Blocks() as app:
                 outputs=[tabs]
             )
 
-
-app.launch()
+app.launch()