translate updated

app.py

@@ -16,9 +16,19 @@ def do_translate(csharp_code, target_format, model_choice):
     class_name = extract_class_name(csharp_code)
     outputs = agent.translate(csharp_code, target_format=target_format, model_choice=model_choice)
 
-    cpp_code = outputs.get("cpp_code", "")
-    header_code = outputs.get("header_code", "")
-    blueprint_code = outputs.get("blueprint_code", "")
+    # Handle if the agent returns tuple instead of dict
+    if isinstance(outputs, tuple):
+        # Assuming structure like (header_code, cpp_code, blueprint_code)
+        if len(outputs) == 3:
+            header_code, cpp_code, blueprint_code = outputs
+        else:
+            header_code, cpp_code, blueprint_code = outputs[0], "", ""
+    elif isinstance(outputs, dict):
+        cpp_code = outputs.get("cpp_code", "")
+        header_code = outputs.get("header_code", "")
+        blueprint_code = outputs.get("blueprint_code", "")
+    else:
+        header_code = cpp_code = blueprint_code = ""
 
     # Save files with class name
     header_path = f"{class_name}.h"
@@ -32,6 +42,7 @@ def do_translate(csharp_code, target_format, model_choice):
     return header_code, cpp_code, blueprint_code, header_path, cpp_path
 
 
+
 title = "Unity → Unreal Code Translator"
 description = """
 ### 🧩 Unity → Unreal Translator