Update app.py

app.py

@@ -1,5 +1,6 @@
 import gradio as gr
 import ast
+import astor
 import os
 from io import StringIO
 
@@ -19,35 +20,54 @@ def parse_gradio_code(code):
         tree = ast.parse(code)
         components = []
         functions = []
+        imports = []
         
         for node in ast.walk(tree):
+            # Extract function definitions
             if isinstance(node, ast.FunctionDef):
-                functions.append(node.name)
+                functions.append(node)
+            # Extract Gradio component calls
             if isinstance(node, ast.Call) and hasattr(node.func, 'attr'):
                 if node.func.attr in GRADIO_TO_TOGA:
                     component = node.func.attr
                     args = [ast.dump(arg, annotate_fields=False) for arg in node.args]
                     kwargs = {kw.arg: ast.dump(kw.value, annotate_fields=False) for kw in node.keywords}
                     components.append({"type": component, "args": args, "kwargs": kwargs})
+            # Extract imports
+            if isinstance(node, (ast.Import, ast.ImportFrom)):
+                imports.append(node)
         
-        return components, functions
+        return components, functions, imports
     except Exception as e:
-        return [], [f"Error parsing code: {str(e)}"]
+        return [], [], [f"Error parsing code: {str(e)}"]
 
 # Generate Toga code
-def generate_toga_code(components, functions):
+def generate_toga_code(components, functions, imports):
     toga_code = [
         "import toga",
         "from toga.style import Pack",
         "from toga.style.pack import COLUMN",
-        "",
-        "# Original functions (copy these from your Gradio app)",
-        f"# Detected functions: {', '.join(functions) if functions else 'None'}",
-        "",
+        ""
+    ]
+    
+    # Add imports
+    toga_code.append("# Dependencies from Gradio app")
+    for imp in imports:
+        toga_code.append(astor.to_source(imp).strip())
+    toga_code.append("")
+    
+    # Add function definitions
+    toga_code.append("# Functions from Gradio app")
+    for func in functions:
+        toga_code.append(astor.to_source(func).strip())
+    toga_code.append("")
+    
+    # Generate UI
+    toga_code.extend([
         "def build(app):",
         "    box = toga.Box(style=Pack(direction=COLUMN, padding=10))",
         ""
-    ]
+    ])
     
     for idx, comp in enumerate(components):
         toga_comp, extra, label_prefix = GRADIO_TO_TOGA.get(comp["type"], ("toga.Label", f"Placeholder: {comp['type']} not supported", "Unknown"))
@@ -77,9 +97,11 @@ def generate_toga_code(components, functions):
         
         elif comp["type"] == "Button":
             label = comp["kwargs"].get("value", f"'{label_prefix} {idx}'")
+            # Check for associated function in click events
+            click_fn = comp["kwargs"].get("fn", f"{functions[0].name if functions else 'function'}")
             toga_code.append(f"    def button_handler_{idx}(widget):")
-            toga_code.append(f"        # Implement logic for {functions[0] if functions else 'function'}")
-            toga_code.append(f"        pass")
+            toga_code.append(f"        # Call {click_fn} with inputs")
+            toga_code.append(f"        pass  # Implement logic using input values")
             toga_code.append(f"    button_{idx} = {toga_comp}(label={label}, {extra})")
             toga_code.append(f"    box.add(button_{idx})")
         
@@ -113,12 +135,12 @@ def convert_gradio_to_toga(file, code_text):
     try:
         # Use file if provided, else use text
         code = file.decode('utf-8') if file else code_text
-        components, functions = parse_gradio_code(code)
+        components, functions, imports = parse_gradio_code(code)
         
         if not components and functions and isinstance(functions[0], str):
             return None, None, f"Error: {functions[0]}"
         
-        toga_code = generate_toga_code(components, functions)
+        toga_code = generate_toga_code(components, functions, imports)
         output_path = "/tmp/toga_app.py"
         with open(output_path, "w") as f:
             f.write(toga_code)