Update app.py

app.py

@@ -1,599 +1,79 @@
 import gradio as gr
 from gradio_client import Client
-import json
-import random
 import time
-from datetime import datetime, timedelta
-import threading
+import random
 
-# Backend engines
-BACKEND_ENGINES = [
-    "ChocoLaboratory/SANDBOXBACKEND2",
-    "ChocoLaboratory/SANDBOX_BACKEND"
+# -------------------------
+# Backend HF clients
+# -------------------------
+backend_engines = [
+    Client("ChocoLaboratory/SANDBOX_BACKEND1"),
+    Client("ChocoLaboratory/SANDBOX_BACKEND2")
 ]
 
-class SandboxManager:
-    def __init__(self):
-        self.client = None
-        self.current_engine = None
-        self.is_connected = False
-        self.sandbox_created = False
-        self.creation_time = None
-        self.time_limit_hours = 2
-        self.auto_refresh_thread = None
-        self.stop_auto_refresh = False
-        
-    def select_random_engine(self):
-        """Select a random backend engine"""
-        self.current_engine = random.choice(BACKEND_ENGINES)
-        return self.current_engine
-        
-    def get_remaining_time(self):
-        """Calculate remaining time for the sandbox"""
-        if not self.creation_time:
-            return "N/A"
-        
-        elapsed = datetime.now() - self.creation_time
-        remaining = timedelta(hours=self.time_limit_hours) - elapsed
-        
-        if remaining.total_seconds() <= 0:
-            return "EXPIRED"
-        
-        hours, remainder = divmod(int(remaining.total_seconds()), 3600)
-        minutes, seconds = divmod(remainder, 60)
-        return f"{hours:02d}:{minutes:02d}:{seconds:02d}"
-    
-    def is_sandbox_expired(self):
-        """Check if sandbox has exceeded the 2-hour limit"""
-        if not self.creation_time:
-            return False
-        
-        elapsed = datetime.now() - self.creation_time
-        return elapsed.total_seconds() > (self.time_limit_hours * 3600)
-        
-    def connect(self):
-        """Connect to a backend engine"""
-        if self.sandbox_created:
-            return "⚠️ Sandbox already created. You can only create one sandbox per session.", self.current_engine
-            
-        engine = self.select_random_engine()
-        try:
-            self.client = Client(engine)
-            self.is_connected = True
-            return f"✅ Connected to: {engine}", engine
-        except Exception as e:
-            self.is_connected = False
-            self.current_engine = None
-            return f"❌ Connection failed to {engine}: {str(e)}", "Connection Failed"
-    
-    def launch_sandbox(self, main_py_code, requirements_txt):
-        """Launch the sandbox (only once per session)"""
-        if self.sandbox_created:
-            remaining = self.get_remaining_time()
-            if remaining == "EXPIRED":
-                return "⏰ Your sandbox has expired (2-hour limit reached). Please refresh the page to start a new session.", ""
-            return f"⚠️ Sandbox already running. Remaining time: {remaining}", self.get_status()
-            
-        if not self.is_connected:
-            return "❌ Not connected to backend. Please connect first.", ""
-            
-        try:
-            result = self.client.predict(
-                code=main_py_code,
-                requirements=requirements_txt,
-                api_name="/launch_sandbox"
-            )
-            
-            # Mark sandbox as created and record creation time
-            self.sandbox_created = True
-            self.creation_time = datetime.now()
-            
-            # Start auto-refresh thread for time monitoring
-            self.start_time_monitor()
-            
-            remaining = self.get_remaining_time()
-            return (f"🚀 Sandbox deployed successfully on {self.current_engine}\n"
-                   f"⏱️  Time limit: 2 hours | Remaining: {remaining}\n\n"
-                   f"Response: {result}"), self.get_status()
-        except Exception as e:
-            return f"❌ Deployment failed: {str(e)}", ""
-    
-    def fetch_logs(self):
-        """Fetch logs from the sandbox"""
-        if not self.is_connected:
-            return "❌ Not connected to backend"
-            
-        if self.is_sandbox_expired():
-            return "⏰ Sandbox expired. Logs are no longer available."
-            
-        try:
-            result = self.client.predict(api_name="/fetch_logs")
-            timestamp = datetime.now().strftime("%H:%M:%S")
-            remaining = self.get_remaining_time()
-            return f"[{timestamp}] Remaining time: {remaining}\nLogs from {self.current_engine}:\n\n{result}"
-        except Exception as e:
-            return f"[ERROR] Failed to fetch logs: {str(e)}"
-    
-    def kill_sandbox(self):
-        """Terminate the sandbox"""
-        if not self.is_connected:
-            return "❌ Not connected to backend", ""
-            
-        try:
-            result = self.client.predict(api_name="/kill_sandbox")
-            self.sandbox_created = False
-            self.creation_time = None
-            self.stop_auto_refresh = True
-            return f"🛑 Sandbox terminated on {self.current_engine}\n\nResponse: {result}", "TERMINATED"
-        except Exception as e:
-            return f"❌ Termination failed: {str(e)}", ""
-    
-    def get_status(self):
-        """Get current sandbox status"""
-        if not self.is_connected:
-            return "OFFLINE"
-            
-        if self.is_sandbox_expired():
-            return "EXPIRED"
-            
-        try:
-            result = self.client.predict(api_name="/status_sandbox")
-            if "running" in str(result).lower():
-                remaining = self.get_remaining_time()
-                return f"RUNNING ({remaining} left) - {result}"
-            else:
-                remaining = self.get_remaining_time()
-                return f"IDLE ({remaining} left) - {result}"
-        except Exception as e:
-            return f"ERROR - {str(e)}"
-    
-    def start_time_monitor(self):
-        """Start monitoring thread for time limits"""
-        def monitor():
-            while not self.stop_auto_refresh and self.sandbox_created:
-                if self.is_sandbox_expired():
-                    try:
-                        self.kill_sandbox()
-                    except:
-                        pass
-                    break
-                time.sleep(60)  # Check every minute
-                
-        self.auto_refresh_thread = threading.Thread(target=monitor, daemon=True)
-        self.auto_refresh_thread.start()
-
-# Initialize sandbox manager
-sandbox = SandboxManager()
-
-# Sample files
-SAMPLE_MAIN_PY = """# Enterprise Python Application
-import os
-import time
-from datetime import datetime
-
-def main():
-    print("🚀 Sandbox Compute Platform - Initialized")
-    print(f"📅 Started at: {datetime.now()}")
-    print("✅ Environment ready for development")
-    
-    # Your application code here
-    while True:
-        print(f"⏱️  Running... {datetime.now().strftime('%H:%M:%S')}")
-        time.sleep(30)
-
-if __name__ == "__main__":
-    main()
-"""
-
-SAMPLE_REQUIREMENTS_TXT = """# Production Dependencies
-# Specify exact versions for reproducibility
-
-# Web framework
-flask==2.3.3
-requests==2.31.0
-
-# Data processing
-pandas==2.1.1
-numpy==1.24.3
-
-# Utilities
-python-dotenv==1.0.0
-"""
-
-# Enhanced CSS with modern design
-industrial_css = """
-@import url('https://fonts.googleapis.com/css2?family=Inter:wght@300;400;500;600;700;800&family=JetBrains+Mono:wght@400;500;600&display=swap');
-
-:root {
-    --primary: #2563eb;
-    --primary-hover: #1d4ed8;
-    --secondary: #64748b;
-    --success: #10b981;
-    --warning: #f59e0b;
-    --danger: #ef4444;
-    --surface: #ffffff;
-    --surface-elevated: #f8fafc;
-    --border: #e2e8f0;
-    --text: #0f172a;
-    --text-muted: #64748b;
-    --shadow: rgba(15, 23, 42, 0.1);
-}
-
-* {
-    font-family: 'Inter', -apple-system, BlinkMacSystemFont, sans-serif !important;
-}
-
-.gradio-container {
-    background: linear-gradient(135deg, #f8fafc 0%, #f1f5f9 100%) !important;
-    min-height: 100vh;
-}
-
-/* Header */
-.platform-header {
-    background: linear-gradient(135deg, var(--primary) 0%, var(--primary-hover) 100%) !important;
-    color: white !important;
-    padding: 2rem !important;
-    border-radius: 0 0 1rem 1rem !important;
-    margin-bottom: 1.5rem !important;
-    box-shadow: 0 10px 25px var(--shadow) !important;
-}
-
-.platform-title {
-    font-size: 2rem !important;
-    font-weight: 800 !important;
-    margin-bottom: 0.5rem !important;
-}
-
-.platform-subtitle {
-    font-size: 1rem !important;
-    opacity: 0.9 !important;
-    font-weight: 400 !important;
-}
-
-.breadcrumb {
-    background: var(--surface-elevated) !important;
-    padding: 0.75rem 1.5rem !important;
-    border-radius: 0.75rem !important;
-    font-size: 0.875rem !important;
-    color: var(--text-muted) !important;
-    border: 1px solid var(--border) !important;
-    margin-bottom: 1.5rem !important;
-}
-
-/* Cards */
-.card {
-    background: var(--surface) !important;
-    border: 1px solid var(--border) !important;
-    border-radius: 1rem !important;
-    box-shadow: 0 4px 6px -1px var(--shadow) !important;
-    overflow: hidden !important;
-    margin-bottom: 1.5rem !important;
-}
-
-.card-header {
-    background: var(--surface-elevated) !important;
-    padding: 1rem 1.5rem !important;
-    border-bottom: 1px solid var(--border) !important;
-    font-weight: 600 !important;
-    font-size: 1.1rem !important;
-    display: flex !important;
-    align-items: center !important;
-    gap: 0.5rem !important;
-}
-
-.card-content {
-    padding: 1.5rem !important;
-}
-
-/* Buttons */
-.btn {
-    padding: 0.75rem 1.5rem !important;
-    border-radius: 0.5rem !important;
-    font-weight: 500 !important;
-    font-size: 0.875rem !important;
-    transition: all 0.2s ease !important;
-    border: none !important;
-}
-
-.btn-primary {
-    background: linear-gradient(135deg, var(--primary), var(--primary-hover)) !important;
-    color: white !important;
-    box-shadow: 0 4px 14px 0 rgba(37, 99, 235, 0.25) !important;
-}
-
-.btn-primary:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 6px 20px 0 rgba(37, 99, 235, 0.35) !important;
-}
-
-.btn-secondary {
-    background: var(--surface) !important;
-    color: var(--text) !important;
-    border: 1px solid var(--border) !important;
-}
-
-.btn-secondary:hover {
-    background: var(--surface-elevated) !important;
-    border-color: var(--secondary) !important;
-}
-
-.btn-danger {
-    background: linear-gradient(135deg, var(--danger), #dc2626) !important;
-    color: white !important;
-    box-shadow: 0 4px 14px 0 rgba(239, 68, 68, 0.25) !important;
-}
-
-.btn-danger:hover {
-    transform: translateY(-2px) !important;
-    box-shadow: 0 6px 20px 0 rgba(239, 68, 68, 0.35) !important;
-}
-
-/* Status indicators */
-.status-badge {
-    padding: 0.5rem 1rem !important;
-    border-radius: 0.5rem !important;
-    font-weight: 500 !important;
-    font-size: 0.875rem !important;
-    text-align: center !important;
-}
-
-.status-online {
-    background: #d1fae5 !important;
-    color: var(--success) !important;
-    border: 1px solid #a7f3d0 !important;
-}
-
-.status-offline {
-    background: #fee2e2 !important;
-    color: var(--danger) !important;
-    border: 1px solid #fecaca !important;
-}
-
-.status-warning {
-    background: #fef3c7 !important;
-    color: var(--warning) !important;
-    border: 1px solid #fed7aa !important;
-}
-
-/* Code editors */
-.code-editor {
-    font-family: 'JetBrains Mono', monospace !important;
-    background: #1e293b !important;
-    color: #e2e8f0 !important;
-    border-radius: 0.75rem !important;
-    border: 1px solid #334155 !important;
-}
-
-.requirements-editor {
-    font-family: 'JetBrains Mono', monospace !important;
-    background: #27272a !important;
-    color: #f4f4f5 !important;
-    border-radius: 0.75rem !important;
-    border: 1px solid #3f3f46 !important;
-}
-
-.terminal {
-    font-family: 'JetBrains Mono', monospace !important;
-    background: #0f172a !important;
-    color: #e2e8f0 !important;
-    border-radius: 0.75rem !important;
-    padding: 1rem !important;
-    border: 1px solid #1e293b !important;
-}
-
-/* Form labels */
-.form-label {
-    font-weight: 600 !important;
-    color: var(--text) !important;
-    margin-bottom: 0.5rem !important;
-    font-size: 0.875rem !important;
-    text-transform: uppercase !important;
-    letter-spacing: 0.025em !important;
-}
-
-/* Monitoring panel */
-.monitor-panel {
-    background: linear-gradient(135deg, #f8fafc 0%, #e2e8f0 100%) !important;
-    border: 1px solid var(--border) !important;
-    border-radius: 1rem !important;
-    padding: 1.5rem !important;
-    box-shadow: 0 4px 6px -1px var(--shadow) !important;
-}
-
-/* Animations */
-@keyframes pulse {
-    0%, 100% { opacity: 1; }
-    50% { opacity: 0.5; }
-}
-
-.pulse {
-    animation: pulse 2s cubic-bezier(0.4, 0, 0.6, 1) infinite;
-}
-"""
-
-def auto_refresh_status():
-    return sandbox.get_status()
-
-def auto_refresh_logs():
-    return sandbox.fetch_logs()
-
-def load_sample_files():
-    return SAMPLE_MAIN_PY, SAMPLE_REQUIREMENTS_TXT
-
-def get_session_info():
-    if sandbox.sandbox_created:
-        remaining = sandbox.get_remaining_time()
-        created = sandbox.creation_time.strftime("%H:%M:%S") if sandbox.creation_time else "N/A"
-        return f"Created: {created} | Remaining: {remaining}"
-    return "No active sandbox"
-
-# Build enhanced Gradio interface
-with gr.Blocks(css=industrial_css, title="Sandbox Compute Platform") as demo:
-    gr.HTML("""
-    <div class="platform-header">
-        <div class="platform-title">🚀 Sandbox Compute Platform</div>
-        <div class="platform-subtitle">Enterprise-grade cloud computing with 2-hour session limits</div>
-    </div>
-    <div class="breadcrumb">
-        🛠️ Sandbox Management Console
-    </div>
-    """)
-
-    # Session Info Bar
+# -------------------------
+# Helper functions
+# -------------------------
+def pick_engine():
+    # Simple load balancing (random for now)
+    return random.choice(backend_engines)
+
+def launch_sandbox(code):
+    engine = pick_engine()
+    result = engine.predict(code=code, api_name="/launch_sandbox")
+    sandbox_id = result if isinstance(result, str) else result.get("sandbox_id", "unknown")
+    return f"Sandbox launched! ID: {sandbox_id}", sandbox_id, "120:00"  # 2h TTL
+
+def run_command(cmd, sandbox_id):
+    engine = pick_engine()
+    # Here we assume the backend runs the code immediately in the sandbox
+    # For demo, we send each command as a mini script
+    code = cmd
+    engine.predict(code=code, api_name="/launch_sandbox")  # you could create a /run_command endpoint
+    logs = engine.predict(api_name="/fetch_logs")
+    return logs
+
+def kill_sandbox(sandbox_id):
+    engine = pick_engine()
+    result = engine.predict(api_name="/kill_sandbox")
+    return result
+
+def get_status(sandbox_id):
+    engine = pick_engine()
+    return engine.predict(api_name="/status_sandbox")
+
+# -------------------------
+# Gradio UI
+# -------------------------
+with gr.Blocks() as demo:
     with gr.Row():
-        with gr.Column():
-            session_info = gr.Textbox(
-                value=get_session_info(),
-                label="📊 Session Information",
-                interactive=False,
-                elem_classes=["status-badge"]
-            )
-
-    with gr.Row():
-        # Main Content
-        with gr.Column(scale=2):
-            # Connection Management
-            with gr.Group(elem_classes=["card"]):
-                gr.HTML('<div class="card-header">🔌 Connection Management</div>')
-                with gr.Group(elem_classes=["card-content"]):
-                    gr.HTML('<div class="form-label">Initialize Backend Connection</div>')
-                    with gr.Row():
-                        connect_btn = gr.Button("🚀 Connect to Backend", elem_classes=["btn-primary"])
-                        load_samples_btn = gr.Button("📄 Load Templates", elem_classes=["btn-secondary"])
-                    
-                    gr.HTML('<div class="form-label">Active Engine</div>')
-                    current_engine_display = gr.Textbox(
-                        value="No engine selected", 
-                        interactive=False,
-                        elem_classes=["status-badge"],
-                        show_label=False
-                    )
-                    
-                    gr.HTML('<div class="form-label">Connection Status</div>')
-                    connection_status = gr.Textbox(
-                        value="DISCONNECTED", 
-                        interactive=False,
-                        elem_classes=["status-offline"],
-                        show_label=False
-                    )
-
-            # Development Environment
-            with gr.Group(elem_classes=["card"]):
-                gr.HTML('<div class="card-header">💻 Development Environment</div>')
-                with gr.Group(elem_classes=["card-content"]):
-                    with gr.Tab("📝 main.py"):
-                        gr.HTML('<div class="form-label">Application Source Code</div>')
-                        main_py_editor = gr.Code(
-                            value=SAMPLE_MAIN_PY,
-                            language="python",
-                            lines=15,
-                            elem_classes=["code-editor"],
-                            show_label=False
-                        )
-                    
-                    with gr.Tab("📦 requirements.txt"):
-                        gr.HTML('<div class="form-label">Dependency Specifications</div>')
-                        requirements_editor = gr.Textbox(
-                            value=SAMPLE_REQUIREMENTS_TXT,
-                            lines=15,
-                            elem_classes=["requirements-editor"],
-                            show_label=False,
-                            max_lines=20
-                        )
-                    
-                    gr.HTML('<hr style="margin: 1.5rem 0; border: 1px solid var(--border);">')
-                    
-                    with gr.Row():
-                        launch_btn = gr.Button("🚀 Deploy Sandbox", elem_classes=["btn-primary"], size="lg")
-                        kill_btn = gr.Button("🛑 Terminate Sandbox", elem_classes=["btn-danger"], size="lg")
-                    
-                    gr.HTML('<div class="form-label">Deployment Output</div>')
-                    launch_output = gr.Textbox(
-                        lines=6,
-                        interactive=False,
-                        placeholder="🚀 Deployment logs and status information will appear here...",
-                        elem_classes=["terminal"],
-                        show_label=False
-                    )
-
-        # Monitoring Panel
+        with gr.Column(scale=3):
+            gr.Markdown("## 🖥 Sandbox Cloud")
+            terminal_output = gr.Textbox(label="Terminal Output", lines=20, interactive=False, elem_classes=["terminal-box"])
+            command_input = gr.Textbox(label="Enter command", placeholder="ls -la", interactive=True)
+            run_btn = gr.Button("Run Command")
         with gr.Column(scale=1):
-            with gr.Group(elem_classes=["monitor-panel"]):
-                gr.HTML('<div class="form-label">📊 System Monitor</div>')
-                
-                gr.HTML('<div class="form-label">Instance Status</div>')
-                status_display = gr.Textbox(
-                    value="OFFLINE",
-                    interactive=False,
-                    elem_classes=["status-offline"],
-                    show_label=False
-                )
-                
-                status_refresh_btn = gr.Button("🔄 Refresh Status", elem_classes=["btn-secondary"], size="sm")
-                
-                gr.HTML('<div class="form-label">⚠️ Session Limits</div>')
-                gr.HTML("""
-                <div style="background: #fef3c7; padding: 1rem; border-radius: 0.5rem; border: 1px solid #fed7aa;">
-                    <strong>Time Limit:</strong> 2 hours per sandbox<br>
-                    <strong>Sessions:</strong> 1 sandbox per browser session<br>
-                    <strong>Auto-cleanup:</strong> Sandbox terminates when expired
-                </div>
-                """)
-
-    # Logs Section
-    with gr.Group(elem_classes=["card"]):
-        gr.HTML('<div class="card-header">📋 System Logs & Monitoring</div>')
-        with gr.Group(elem_classes=["card-content"]):
-            with gr.Row():
-                fetch_logs_btn = gr.Button("📋 Fetch Application Logs", elem_classes=["btn-secondary"])
-                auto_refresh_btn = gr.Button("🔄 Auto Refresh (30s)", elem_classes=["btn-secondary"])
-            
-            gr.HTML('<div class="form-label">Runtime Logs</div>')
-            logs_display = gr.Textbox(
-                lines=12,
-                interactive=False,
-                elem_classes=["terminal"],
-                placeholder="[SYSTEM] 📋 Application logs will appear here...",
-                show_label=False
-            )
-
-    # Event Handlers
-    connect_btn.click(
-        fn=sandbox.connect,
-        outputs=[connection_status, current_engine_display]
-    )
-    
-    load_samples_btn.click(
-        fn=load_sample_files,
-        outputs=[main_py_editor, requirements_editor]
-    )
-    
-    launch_btn.click(
-        fn=sandbox.launch_sandbox,
-        inputs=[main_py_editor, requirements_editor],
-        outputs=[launch_output, status_display]
-    )
-    
-    kill_btn.click(
-        fn=sandbox.kill_sandbox,
-        outputs=[launch_output, status_display]
-    )
-    
-    fetch_logs_btn.click(
-        fn=sandbox.fetch_logs,
-        outputs=logs_display
-    )
-    
-    status_refresh_btn.click(
-        fn=auto_refresh_status,
-        outputs=status_display
-    )
+            launch_btn = gr.Button("Launch Sandbox")
+            kill_btn = gr.Button("Kill Sandbox")
+            status_btn = gr.Button("Check Status")
+            sandbox_id_box = gr.Textbox(label="Sandbox ID", interactive=False)
+            time_remaining = gr.Textbox(label="Time Remaining", interactive=False)
+            gr.Markdown("💡 Note: Running on a Debian-based container!")
+
+    # Callbacks
+    launch_btn.click(launch_sandbox, inputs=command_input, outputs=[terminal_output, sandbox_id_box, time_remaining])
+    run_btn.click(run_command, inputs=[command_input, sandbox_id_box], outputs=terminal_output)
+    kill_btn.click(kill_sandbox, inputs=sandbox_id_box, outputs=terminal_output)
+    status_btn.click(get_status, inputs=sandbox_id_box, outputs=terminal_output)
+
+# CSS for terminal look
+css = """
+.terminal-box textarea {
+    background-color: #1e1e1e;
+    color: #00ff00;
+    font-family: monospace;
+    font-size: 14px;
+}
+"""
 
-# Launch the application
-if __name__ == "__main__":
-    demo.launch(
-        server_name="0.0.0.0",
-        server_port=7860,
-        share=False,
-        debug=False
-    )
+demo.launch(css=css)