Add validation for tflite models uploads

app.py

@@ -4,11 +4,47 @@ import tempfile
 import os
 import sr100_model_compiler
 import html
+import pathlib
 
-def compile_model(model_name, vmem_value, lpmem_value):
+# ---------- Helpers ----------
 
-    #if oauth_info['token'] is None:
-    #    return "ERROR - please log into HuggingFace to continue"
+def _resolve_uploaded_path(uploaded):
+    """
+    Normalize Gradio File input into a filesystem path.
+    Handles: str, dict with {path|name}, file-like objects with .path/.name,
+    or a list/tuple of the above.
+    """
+    if uploaded is None:
+        return None
+    if isinstance(uploaded, (list, tuple)) and uploaded:
+        return _resolve_uploaded_path(uploaded[0])
+    if isinstance(uploaded, str):
+        return uploaded
+    if isinstance(uploaded, dict):
+        return uploaded.get("path") or uploaded.get("name")
+    for attr in ("path", "name"):
+        if hasattr(uploaded, attr):
+            return getattr(uploaded, attr)
+    return None
+
+def compile_model(model_name, vmem_value, lpmem_value, uploaded_model):
+    # Decide the source model path (uploaded has priority)
+    uploaded_path = _resolve_uploaded_path(uploaded_model)
+    model_path = uploaded_path or model_name
+
+    # Basic validations
+    if not model_path or not os.path.exists(model_path):
+        return (
+            "<div style='color:#d32f2f; font-weight:bold; font-size:1.1em;'>"
+            "❌ ERROR: Could not locate the model file you selected or uploaded."
+            "</div>"
+        )
+
+    if pathlib.Path(model_path).suffix.lower() != ".tflite":
+        return (
+            "<div style='color:#d32f2f; font-weight:bold; font-size:1.1em;'>"
+            "❌ ERROR: Please provide a <code>.tflite</code> model file.</div>"
+        )
 
     # Create a temporary directory
     with tempfile.TemporaryDirectory() as out_dir:
@@ -19,13 +55,12 @@ def compile_model(model_name, vmem_value, lpmem_value):
         
         # Run the model fitter
         success, results = sr100_model_compiler.sr100_model_optimizer(
-            model_file=model_name,
+            model_file=model_path,
             vmem_size_limit=vmem_size_limit,
             lpmem_size_limit=lpmem_size_limit
         )
         print(results)
 
-    # Format results in nicely styled HTML like in old.py
     output = []
     
     if results['cycles_npu'] == 0:
@@ -97,7 +132,6 @@ def compile_model(model_name, vmem_value, lpmem_value):
 # Get all available models
 model_choices = glob.glob('models/*.tflite')
 
-# Custom CSS from old.py
 custom_css = """
 :root {
     --color-accent: #007dc3;
@@ -140,32 +174,31 @@ footer, .gradio-footer, .svelte-1ipelgc, .gradio-logo, .gradio-app__settings {
 """
     
 with gr.Blocks(css=custom_css) as demo:
-    #gr.LoginButton()
     gr.Markdown("<h1 style='font-size:2.5em; color:#007dc3; margin-bottom:0;'>SR100 Model Compiler</h1>", elem_id="main_title")
     gr.Markdown("<h3 style='margin-top:0; color:#000;'>Bring a TFlite INT8 model and compile it for Synaptics Astra SR100. Learn more at <a href='https://developer.synaptics.com/docs/sr/sr100/quick-start?utm_source=hf' target='_blank' style='color:#007dc3; text-decoration:underline;'>Synaptics AI Developer Zone</a></h3>", elem_id="subtitle")
-    #user_text = gr.Markdown("")
 
     # Setup model inputs
     with gr.Row():
         vmem_slider = gr.Slider(minimum=0, maximum=1536, step=1.024, label="Set total VMEM SRAM size available in kB", value=1536.0)
         lpmem_slider = gr.Slider(minimum=0, maximum=1536, step=1.024, label="Set total LPMEM SRAM size in kB", value=1536.0)
     
-    # Setup model compile
+    # Setup model selection/upload
     model_dropdown = gr.Dropdown(
-        label="Select an model",
+        label="Select a model",
         value='models/hello_world.tflite',
         choices=model_choices
     )
     
+    # Add file upload component
+    model_upload = gr.File(label="Or upload a .tflite INT8 model", file_types=[".tflite"], file_count="single")
+    
     # Run the compile
     compile_btn = gr.Button("Compile Model")
     compile_text = gr.Markdown("<span style='color:#000;'>Waiting for model results</span>")
     
     # Compute options
-    compile_btn.click(compile_model, inputs=[model_dropdown, vmem_slider, lpmem_slider], outputs=[compile_text])
-    #demo.load(get_oauth_info, inputs=None, outputs=user_text)
+    compile_btn.click(compile_model, inputs=[model_dropdown, vmem_slider, lpmem_slider, model_upload], outputs=[compile_text])
     
-    # Add footer content from old.py
     gr.HTML("""
     <div style="max-width: 800px; margin: 2rem auto; background: white; color: black; border-radius: 12px; box-shadow: 0 4px 6px -1px rgba(0, 0, 0, 0.1), 0 2px 4px -1px rgba(0, 0, 0, 0.06); border: 1px solid #e5e7eb; padding: 1.5rem; text-align: center;">
         For a detailed walkthrough, please see our