Update app.py

app.py

@@ -3,18 +3,19 @@ import requests
 import time
 
 # --- CONFIGURATION ---
+# Replace with your actual URLs
 DROPLET_IP = "134.199.195.151" 
 API_URL = "https://djrbe-134-199-192-140.run.pinggy-free.link/translate"
+LABLAB_URL = "https://lablab.ai/event/amd-ai-hackathon/ghost-coder" # Update this link!
 
 # --- LOGIC ---
 def ghost_translate(cuda_code):
     yield "👻 Ghost-Coder: Analyzing CUDA Kernel...", "Loading..."
     
     try:
-        # We use a pure-text marker that the backend cannot delete or mangle
+        # Standard marker for backend slicing
         split_marker = "// --- GHOST-CODER HIP OUTPUT ---"
         
-        # A standard completion prompt that forces the model to finish the file
         formatted_prompt = f"""Task: Translate the following CUDA code to AMD HIP. 
 Output ONLY valid C++ code. Do not include markdown blocks or explanations.
 
@@ -29,35 +30,39 @@ Output ONLY valid C++ code. Do not include markdown blocks or explanations.
         if response.status_code == 200:
             raw_response = response.json().get("hip_code", "")
             
-            # THE HARD SLICE: Cut away the entire input prompt
+            # Hard Slice logic to ensure clean output
             if split_marker in raw_response:
                 hip_code = raw_response.split(split_marker)[-1].strip()
             else:
-                # Failsafe if the model acts weird
                 hip_code = raw_response.strip()
             
-            # Step 3: Agentic Visual Steps
+            # Agentic Visual Feedback
             yield "🔄 Analyzing HIP logic on ROCm stack...", "Generating..."
             time.sleep(1)
             yield "🛠️ Verifying syntax and memory offsets...", "Verifying..."
             time.sleep(1)
             yield "✅ Self-Healing successful! HIP Code generated.", hip_code
         else:
-            yield f"❌ Droplet Error: {response.status_code}", "// Check logs"
+            yield f"❌ Droplet Error: {response.status_code}", "// Check bridge logs on MI300X"
             
     except Exception as e:
         yield f"❌ Connection Error: Ensure bridge is running", str(e)
 
-with gr.Blocks() as demo:
-    gr.HTML("""
-        <div style="background-color: #fff3cd; color: #856404; padding: 15px; border-radius: 5px; border: 1px solid #ffeeba; margin-bottom: 20px;">
-            <strong>⚠️ Demo Status: GPU Backend Paused</strong><br>
-            The AMD MI300X instance for this live demo has been paused due to the conclusion of the hackathon credit period. 
-            To see the agent in action, please view the <strong><a href="YOUR_LABLAB_SUBMISSION_URL" target="_blank">Full Video Presentation on Lablab.ai</a></strong>.
-        </div>
-    """)
+# --- UI DESIGN ---
+with gr.Blocks(css=".banner { margin-bottom: 20px; }") as demo:
+    
+    # Header Banner - Wrapped for maximum compatibility
+    with gr.Row():
+        gr.HTML(f"""
+            <div style="background-color: #fff3cd; color: #856404; padding: 20px; border-radius: 8px; border: 1px solid #ffeeba; width: 100%; text-align: center; font-family: sans-serif;">
+                <span style="font-size: 1.2em;">⚠️ <strong>Demo Status: GPU Backend Paused</strong></span><br>
+                <p style="margin-top: 10px;">The AMD MI300X instance for this live demo has been paused following the conclusion of the hackathon credit period.</p>
+                <a href="{LABLAB_URL}" target="_blank" style="display: inline-block; background-color: #856404; color: white; padding: 10px 20px; text-decoration: none; border-radius: 5px; font-weight: bold; margin-top: 5px;">View Full Video Demo & Technical Pitch</a>
+            </div>
+        """)
+
     gr.Markdown("# 👻 Ghost-Coder: Autonomous CUDA-to-HIP Agent")
-    gr.Markdown("### Powered by AMD Instinct™ MI300X | Qwen2.5-Coder-32B")
+    gr.Markdown("### Optimized for AMD Instinct™ MI300X | Qwen2.5-Coder-32B")
     
     with gr.Row():
         with gr.Column():
@@ -65,7 +70,7 @@ with gr.Blocks() as demo:
                 label="Paste CUDA Code Here", 
                 language="cpp", 
                 lines=15, 
-                value="// Example CUDA Kernel\n__global__ void add(int *a) { ... }"
+                value="// Example CUDA Host Code\nvoid runKernel() {\n    cudaMalloc(&d_A, size);\n    kernel<<<256, 256>>>(d_A);\n}"
             )
             run_btn = gr.Button("Translate & Verify", variant="primary")
         
@@ -73,6 +78,9 @@ with gr.Blocks() as demo:
             output_code = gr.Code(label="Generated HIP Code", language="cpp", lines=15)
             logs = gr.Textbox(label="Agent Status & Self-Healing Logs", interactive=False)
 
+    # UI Action
     run_btn.click(ghost_translate, inputs=[input_code], outputs=[logs, output_code])
 
-demo.queue().launch(theme=gr.themes.Soft())
+# Launch configuration
+if __name__ == "__main__":
+    demo.queue().launch(show_api=False)