update: submit file

src/submission/submit.py

@@ -1,11 +1,10 @@
+import gradio as gr
 import json
 import os
 from datetime import datetime, timezone
-
 from src.display.formatting import styled_error, styled_message
 from src.envs import API, EVAL_REQUESTS_PATH, QUEUE_REPO
 
-
 def add_new_eval(model: str, weight_type: str, gguf_filename=None):
     user_name = ""
     model_path = model
@@ -21,15 +20,12 @@ def add_new_eval(model: str, weight_type: str, gguf_filename=None):
     except Exception:
         return styled_error("Could not get your model information.")
     
-    # Block gguf_filename input if weight_type is safetensors
     if weight_type == "safetensors":
         if gguf_filename:
             return styled_error("GGUF filename should not be provided when using safetensors.")
         gguf_filename = None  # Ensure gguf_filename is None when safetensors is selected
 
-    # Seems good, creating the eval
-    print("Adding new eval")
-
+    # Create the eval entry
     eval_entry = {
         "model": model,
         "weight_type": weight_type,
@@ -38,7 +34,7 @@ def add_new_eval(model: str, weight_type: str, gguf_filename=None):
         "submitted_time": current_time,
     }
 
-    print("Creating eval file")
+    # Create output directory and file
     OUT_DIR = f"{EVAL_REQUESTS_PATH}/{user_name}"
     os.makedirs(OUT_DIR, exist_ok=True)
     out_path = f"{OUT_DIR}/{model_path}_eval_request_{current_time}.json"
@@ -46,7 +42,7 @@ def add_new_eval(model: str, weight_type: str, gguf_filename=None):
     with open(out_path, "w") as f:
         f.write(json.dumps(eval_entry))
 
-    print("Uploading eval file")
+    # Upload eval file
     API.upload_file(
         path_or_fileobj=out_path,
         path_in_repo=out_path.split("eval-queue/")[1],
@@ -61,3 +57,31 @@ def add_new_eval(model: str, weight_type: str, gguf_filename=None):
     return styled_message(
         "Your request has been submitted to the evaluation queue!\nPlease wait for up to five minutes for the model to show in the PENDING list."
     )
+
+def update_gguf_input(weight_type):
+    return gr.update(interactive=weight_type != "safetensors")
+
+# Gradio interface
+with gr.Blocks() as demo:
+    model_input = gr.Textbox(label="Model")
+    weight_type_input = gr.Dropdown(
+        label="Weight Type",
+        choices=["default", "safetensors", "other"],
+        value="default",
+        interactive=True,
+        elem_id="weight-type-dropdown"
+    )
+    gguf_filename_input = gr.Textbox(label="GGUF Filename", interactive=True)
+    submit_btn = gr.Button("Submit")
+    output = gr.Markdown()
+
+    # Update gguf_filename input based on weight_type selection
+    weight_type_input.change(fn=update_gguf_input, inputs=weight_type_input, outputs=gguf_filename_input)
+
+    submit_btn.click(
+        fn=add_new_eval,
+        inputs=[model_input, weight_type_input, gguf_filename_input],
+        outputs=output
+    )
+
+demo.launch()