Upload app.py

app.py

@@ -7,41 +7,54 @@ DATA_DIR = "submissions"
 os.makedirs(DATA_DIR, exist_ok=True)
 
 # Predefined task types
-TASK_TYPES = ["math", "Open-Ended", "Custom"]
+TASK_TYPES = ["Classification", "Regression", "Translation"]
 
 # Function to handle task submission
 def submit_task(task_type, description, yaml_text):
     if not yaml_text.strip():
         return "YAML/Text input cannot be empty."
 
-    # Save task data
+    # Prepare data
     data = {
         "task_type": task_type,
         "description": description,
         "yaml": yaml_text
     }
     file_path = os.path.join(DATA_DIR, f"{task_type}_{len(os.listdir(DATA_DIR))}.json")
-    with open(file_path, "w") as f:
-        json.dump(data, f)
-
-    return f"Task submitted successfully under type '{task_type}'!"
+    
+    try:
+        with open(file_path, "w") as f:
+            json.dump(data, f)
+        return f"Task submitted successfully under type '{task_type}'!"
+    except Exception as e:
+        return f"Error saving task: {e}"
 
 # Function to get tasks by type
 def get_tasks_by_type(task_type):
     tasks = []
     for file in os.listdir(DATA_DIR):
-        with open(os.path.join(DATA_DIR, file), "r") as f:
-            data = json.load(f)
-            if data["task_type"] == task_type:
-                tasks.append(data)
+        # Skip non-JSON files
+        if not file.endswith(".json"):
+            continue
+        
+        try:
+            with open(os.path.join(DATA_DIR, file), "r") as f:
+                data = json.load(f)
+                # Filter by task type
+                if data.get("task_type") == task_type:
+                    tasks.append(data)
+        except (json.JSONDecodeError, KeyError):
+            # Skip invalid or corrupted files
+            print(f"Skipped invalid file: {file}")
     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."
+        # Return the updated choices
+        return TASK_TYPES, f"Task type '{new_type}' added successfully!"
+    return TASK_TYPES, "Task type already exists or invalid input."
 
 # Function to switch tabs
 def switch_tab(tab_name):
@@ -59,7 +72,7 @@ with gr.Blocks() as app:
             task_type_input = gr.Dropdown(
                 label="Task Type",
                 choices=TASK_TYPES,
-                value="math",
+                value="Classification",
                 interactive=True
             )
             new_task_input = gr.Textbox(
@@ -111,7 +124,7 @@ with gr.Blocks() as app:
             task_type_filter = gr.Dropdown(
                 label="Filter by Task Type",
                 choices=TASK_TYPES,
-                value="math",
+                value="Classification",
                 interactive=True
             )
             view_button = gr.Button("View Tasks")
@@ -137,4 +150,4 @@ with gr.Blocks() as app:
                 outputs=[tabs]
             )
 
-app.launch()
+app.launch()