Update deployer/gradio_generator.py

deployer/gradio_generator.py

@@ -4,44 +4,51 @@ import zipfile
 from core_creator.voice_to_app import VoiceToAppCreator
 from deployer.simulator_interface import VirtualRobot
 
+def _make_app_zip(src_dir: str, title: str) -> str:
+    """
+    Zips up the directory at src_dir into a file named <title>.zip
+    in the system temp directory, and returns the full path.
+    """
+    safe_title = "".join(c if c.isalnum() or c in "-_." else "_" for c in title)
+    zip_filename = f"{safe_title}.zip"
+    tmp_dir = tempfile.gettempdir()
+    zip_path = os.path.join(tmp_dir, zip_filename)
+    with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
+        for root, _, files in os.walk(src_dir):
+            for fname in files:
+                full_path = os.path.join(root, fname)
+                # store relative path within the zip
+                arcname = os.path.relpath(full_path, src_dir)
+                zf.write(full_path, arcname)
+    return zip_path
+
 def deploy_callback(idea: str):
     """
-    1) Generate the app assets via VoiceToAppCreator
-    2) Package them into a ZIP under /tmp (tempdir)
-    Returns (status_message, zip_file_path_or_None)
+    Generates an app from the user idea and returns:
+      (status_message: str, path_to_zip: Optional[str])
     """
-    # --- 1) Run pipeline ---
     try:
         creator = VoiceToAppCreator(idea)
-        assets = creator.run_pipeline()  # should return {"blueprint": {...}, "files": {filename:content_str}}
+        assets = creator.run_pipeline()
+        # Title for status
         title = assets.get("blueprint", {}).get("title", "<unknown>")
-    except Exception as e:
-        return f"❌ Failed to generate app: {e}", None
-
-    # --- 2) Package into a ZIP ---
-    try:
-        # create a fresh temp directory
-        base_dir = tempfile.mkdtemp(prefix="robosage_")
-        zip_path = os.path.join(base_dir, "app.zip")
-
-        with zipfile.ZipFile(zip_path, "w", zipfile.ZIP_DEFLATED) as zf:
-            for fname, content in assets.get("files", {}).items():
-                # ensure subdirs exist
-                full_path = os.path.join(base_dir, fname)
-                os.makedirs(os.path.dirname(full_path), exist_ok=True)
-                # write the generated file
-                with open(full_path, "w", encoding="utf-8") as f:
-                    f.write(content)
-                # add to zip under its relative name
-                zf.write(full_path, arcname=fname)
+        # Directory where the app was generated
+        project_dir = assets.get("project_dir") or assets.get("app_dir")
+        if project_dir is None or not os.path.isdir(project_dir):
+            # No directory to zip up
+            return f"❌ No generated directory found for '{title}'", None
 
+        zip_path = _make_app_zip(project_dir, title)
         return f"✅ Generated app: {title}", zip_path
 
     except Exception as e:
-        return f"❌ Failed to package app: {e}", None
+        return f"❌ Failed to generate app: {e}", None
 
-def robot_behavior(user_input: str) -> str:
+def robot_behavior(command: str) -> str:
     """
-    Hook into your simulator.
+    Feeds the user command into the VirtualRobot simulator.
     """
-    return VirtualRobot().perform_action(user_input)
+    try:
+        return VirtualRobot().perform_action(command)
+    except Exception as e:
+        return f"❌ Simulator error: {e}"