Upload app.py

app.py

@@ -1,61 +1,120 @@
-import os
 import gradio as gr
+import os
+import json
+
+# Directory to store submissions
+DATA_DIR = "submissions"
+os.makedirs(DATA_DIR, exist_ok=True)
+
+# Predefined task types
+TASK_TYPES = ["Classification", "Regression", "Translation", "Custom"]
+
+# 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
+    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}'!"
+
+# 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)
+    return tasks
 
-# Directory to store uploaded files
-UPLOAD_DIR = "uploaded_cards"
-os.makedirs(UPLOAD_DIR, exist_ok=True)
-
-# Function to handle file uploads
-def upload_config_card(file, description):
-    if not file:
-        return "Please upload a file."
-    
-    # Save file
-    file_path = os.path.join(UPLOAD_DIR, file.name)
-    with open(file_path, "wb") as f:
-        f.write(file.read())
-    
-    # Save description
-    meta_path = file_path + "_meta.txt"
-    with open(meta_path, "w") as meta_file:
-        meta_file.write(description)
-    
-    return f"File '{file.name}' uploaded successfully!"
-
-# Function to list uploaded files
-def list_uploaded_files():
-    files = []
-    for file in os.listdir(UPLOAD_DIR):
-        if not file.endswith("_meta.txt"):
-            file_path = os.path.join(UPLOAD_DIR, file)
-            meta_path = file_path + "_meta.txt"
-            description = "No description provided."
-            if os.path.exists(meta_path):
-                with open(meta_path, "r") as meta_file:
-                    description = meta_file.read()
-            files.append({"File Name": file, "Description": description, "Download Link": file_path})
-    return files
-
-# Gradio Interface
-def refresh_file_list():
-    return list_uploaded_files()
+# Function to switch tabs
+def switch_tab(tab_name):
+    return gr.Tabs.update(visible_tab=tab_name)
 
+# Gradio App
 with gr.Blocks() as app:
-    gr.Markdown("# Configuration Card Sharing Space")
-    gr.Markdown("Upload and share configuration cards with descriptions of their tasks.")
-
-    with gr.Row():
-        with gr.Column():
-            file_input = gr.File(label="Upload Configuration Card")
-            description_input = gr.Textbox(label="Task Description", placeholder="Describe the task here...")
-            upload_button = gr.Button("Upload")
-            upload_status = gr.Textbox(label="Upload Status", interactive=False)
-        with gr.Column():
-            refresh_button = gr.Button("Refresh List")
-            file_table = gr.Dataframe(headers=["File Name", "Description", "Download Link"], interactive=False)
-
-    # Bind upload and refresh functions
-    upload_button.click(upload_config_card, inputs=[file_input, description_input], outputs=[upload_status])
-    refresh_button.click(refresh_file_list, outputs=[file_table])
+    gr.Markdown("# Task Configuration Sharing Space")
+
+    with gr.Tabs() as tabs:
+        with gr.Tab("Submit Task"):
+            gr.Markdown("## Submit a New Task Configuration")
+            
+            # Input fields for the task submission
+            task_type_input = gr.Dropdown(
+                label="Task Type",
+                choices=TASK_TYPES,
+                value="Classification",
+                interactive=True
+            )
+            description_input = gr.Textbox(
+                label="Task Description",
+                placeholder="Provide a brief description of the task.",
+                lines=3
+            )
+            yaml_input = gr.Textbox(
+                label="YAML/Text Input",
+                placeholder="Paste your YAML or text configuration here.",
+                lines=10
+            )
+            submit_button = gr.Button("Submit Task")
+            go_to_view_tab = gr.Button("Go to View Tasks")
+            submission_status = gr.Textbox(label="Status", interactive=False)
+
+            # Handle task submission
+            submit_button.click(
+                submit_task,
+                inputs=[task_type_input, description_input, yaml_input],
+                outputs=[submission_status]
+            )
+
+            # Button to switch to "View Tasks" tab
+            go_to_view_tab.click(
+                switch_tab,
+                inputs=None,
+                outputs=[tabs],
+                fn_kwargs={"tab_name": "View Tasks"}
+            )
+
+        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",
+                interactive=True
+            )
+            view_button = gr.Button("View Tasks")
+            task_table = gr.Dataframe(headers=["Task Type", "Description", "YAML/Text"], interactive=False)
+            go_to_submit_tab = gr.Button("Go to Submit Task")
+
+            # Function to filter tasks by type
+            def filter_tasks(task_type):
+                tasks = get_tasks_by_type(task_type)
+                return [{"Task Type": t["task_type"], "Description": t["description"], "YAML/Text": t["yaml"]} for t in tasks]
+
+            # Handle task filtering
+            view_button.click(
+                filter_tasks,
+                inputs=[task_type_filter],
+                outputs=[task_table]
+            )
+
+            # Button to switch to "Submit Task" tab
+            go_to_submit_tab.click(
+                switch_tab,
+                inputs=None,
+                outputs=[tabs],
+                fn_kwargs={"tab_name": "Submit Task"}
+            )
 
 app.launch()