app ui updated

app.py

@@ -5,16 +5,11 @@ import tempfile
 from translator.agent import TranslationAgent
 from difflib import unified_diff
 
-# Initialize agent (will load mapping table etc.)
+# Initialize agent
 agent = TranslationAgent()
 
-'''
-def do_translate(csharp_code, target_format, model_choice):
-    # target_format: "cpp", "blueprint", "both"
-    return agent.translate(csharp_code, target_format=target_format, model_choice=model_choice)
-'''
 
-# --- Save cpp files and BP Files ---
+# --- Helper functions ---
 def save_cpp_files(cpp_text: str):
     h_file = tempfile.NamedTemporaryFile(delete=False, suffix=".h")
     cpp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".cpp")
@@ -44,44 +39,37 @@ def show_diff(csharp: str, cpp: str) -> str:
     )
     return "\n".join(diff)
 
+
+def color_diff(diff_text: str) -> str:
+    """Wrap + lines in green, - lines in red for HTML preview"""
+    html = "<pre style='font-family: monospace;'>"
+    for line in diff_text.splitlines():
+        if line.startswith("+"):
+            html += f'<span style="color:green">{line}</span>\n'
+        elif line.startswith("-"):
+            html += f'<span style="color:red">{line}</span>\n'
+        else:
+            html += f"{line}\n"
+    html += "</pre>"
+    return html
+
+
 def translate_with_mapping(csharp_code, target_format, model_choice, mapping_text):
     try:
         mapping = json.loads(mapping_text)
     except json.JSONDecodeError:
-        return "⚠️ Invalid mapping JSON", "", "", ""
-    
-    # Update agent mapping
+        return "Invalid mapping JSON", "", "", ""
+
     agent.mapping = mapping
     cpp, blueprint = agent.translate(csharp_code, target_format=target_format, model_choice=model_choice)
-    diff = show_diff(csharp_code, cpp)
+    diff_html = color_diff(show_diff(csharp_code, cpp))
     h_file, cpp_file = save_cpp_files(cpp)
-    return h_file, cpp_file, blueprint, diff
-
-'''
-title = "Unity → Unreal Code Translator"
-description = "Paste a Unity C# script and get suggested Unreal C++ (.h/.cpp) and an Unreal Python Blueprint-generator script."
-
-iface = gr.Interface(
-    fn=do_translate,
-    inputs=[
-        gr.Textbox(lines=20, label="Unity C# code"),
-        gr.Radio(["cpp", "blueprint", "both"], value="both", label="Target"),
-        gr.Radio(["hf", "openai"], value="hf", label="Model backend")
-    ],
-    outputs=[
-        gr.Textbox(lines=20, label="Unreal C++ (.h/.cpp)"),
-        gr.Textbox(lines=20, label="Unreal Blueprint generator (Python for Editor)")
-    ],
-    title=title,
-    description=description,
-    flagging_mode="never",
-)
-
-'''
+    return h_file, cpp_file, blueprint, diff_html
+
 
 # --- Gradio UI ---
 with gr.Blocks(title="Unity → Unreal Code Translator") as demo:
-    gr.Markdown("# 🧠 Unity → Unreal Code Translator")
+    gr.Markdown("# Unity → Unreal Code Translator")
     gr.Markdown("Paste a Unity C# script and get suggested Unreal C++ (.h/.cpp) and a Blueprint Python generator.")
 
     with gr.Row():
@@ -94,7 +82,7 @@ with gr.Blocks(title="Unity → Unreal Code Translator") as demo:
         cpp_file_out = gr.File(label=".cpp file")
         blueprint_out = gr.Code(label="Blueprint Python", language="python")
 
-    diff_output = gr.Code(label="Preview Diff (C# → C++)", language="diff")
+    diff_output = gr.HTML(label="Preview Diff (C# → C++)")  # HTML for colored diff
 
     with gr.Accordion("Mapping Rules Playground", open=False):
         mapping_editor = gr.Textbox(
@@ -103,7 +91,7 @@ with gr.Blocks(title="Unity → Unreal Code Translator") as demo:
             label="Edit Mapping JSON"
         )
 
-    translate_btn = gr.Button("Translate 🚀")
+    translate_btn = gr.Button("Translate")
     translate_btn.click(
         translate_with_mapping,
         inputs=[csharp_box, target_format, backend, mapping_editor],
@@ -112,4 +100,3 @@ with gr.Blocks(title="Unity → Unreal Code Translator") as demo:
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, debug=True)
-