Fix DownloadButton schema error, upgrade gradio

app.py

@@ -1,6 +1,7 @@
 """dcode Gradio Space - Text to Gcode inference."""
 
 import re
+import tempfile
 import gradio as gr
 import torch
 from transformers import AutoModelForSeq2SeqLM, AutoTokenizer
@@ -128,18 +129,28 @@ def gcode_to_svg(gcode: str) -> str:
 generator = GcodeGenerator()
 
 
-def generate(prompt: str, temperature: float) -> tuple[str, str, str]:
+def generate(prompt: str, temperature: float):
     """Generate gcode from prompt."""
     if not prompt.strip():
-        return "", "", "Enter a prompt"
+        return "", "", "Enter a prompt", None, None
 
     try:
         gcode = generator.generate(prompt, temperature=temperature)
         svg = gcode_to_svg(gcode)
         status = f"✓ Generated {len(gcode.split(chr(10)))} lines, machine compatible"
-        return gcode, svg, status
+        
+        # Create temp files for download
+        gcode_file = tempfile.NamedTemporaryFile(mode='w', suffix='.gcode', delete=False)
+        gcode_file.write(gcode)
+        gcode_file.close()
+        
+        svg_file = tempfile.NamedTemporaryFile(mode='w', suffix='.svg', delete=False)
+        svg_file.write(svg)
+        svg_file.close()
+        
+        return gcode, svg, status, gcode_file.name, svg_file.name
     except Exception as e:
-        return "", "", f"Error: {e}"
+        return "", "", f"Error: {e}", None, None
 
 
 # Custom CSS
@@ -178,12 +189,20 @@ with gr.Blocks(css=custom_css, theme=gr.themes.Base(primary_hue="green")) as dem
         gcode_output = gr.Textbox(label="Gcode", lines=10, show_copy_button=True)
 
     with gr.Row():
-        gr.DownloadButton("Download Gcode", value=lambda: None, visible=False)
+        gcode_download = gr.File(label="Download Gcode")
+        svg_download = gr.File(label="Download SVG")
 
-    generate_btn.click(generate, [prompt, temperature], [gcode_output, preview, status])
-    prompt.submit(generate, [prompt, temperature], [gcode_output, preview, status])
+    generate_btn.click(
+        generate, 
+        [prompt, temperature], 
+        [gcode_output, preview, status, gcode_download, svg_download]
+    )
+    prompt.submit(
+        generate, 
+        [prompt, temperature], 
+        [gcode_output, preview, status, gcode_download, svg_download]
+    )
 
 
 if __name__ == "__main__":
     demo.launch()
-

requirements.txt

@@ -1,5 +1,4 @@
-gradio>=4.0
+gradio>=4.44.1
 torch>=2.0
 transformers>=4.36
 accelerate>=0.25
-