Upload folder using huggingface_hub

server/app.py

@@ -7,8 +7,7 @@
 """
 FastAPI application with Gradio UI for the Quantum Circuit Optimization Environment.
 
-This module creates an HTTP server that exposes the QuantumCircuitEnvironment
-with both OpenEnv API endpoints and a custom Gradio UI.
+This module creates a custom Gradio interface for quantum circuit optimization.
 
 Usage:
     # Development (with auto-reload):
@@ -22,20 +21,10 @@ Usage:
 """
 
 import logging
-import os
-from pathlib import Path
 
 import gradio as gr
 from fastapi import FastAPI
 from fastapi.middleware.cors import CORSMiddleware
-from fastapi.responses import HTMLResponse, RedirectResponse
-
-try:
-    from openenv.core.env_server.http_server import create_app
-except Exception as e:
-    raise ImportError(
-        "openenv is required for the web interface. Install dependencies with 'uv sync'"
-    ) from e
 
 try:
     from ..models import ActionType, GateType, QuantumAction, QuantumObservation
@@ -48,29 +37,12 @@ logging.basicConfig(level=logging.INFO)
 logger = logging.getLogger(__name__)
 
 # ---------------------------------------------------------------------------
-# Environment factory
-# ---------------------------------------------------------------------------
-
-def make_env():
-    """Factory function to create a new environment instance."""
-    env = QuantumCircuitEnvironment(seed=42)
-    env.reset(task_id="easy")
-    return env
-
-
-# ---------------------------------------------------------------------------
-# Create OpenEnv FastAPI app
+# FastAPI app
 # ---------------------------------------------------------------------------
 
-app = create_app(
-    make_env,
-    QuantumAction,
-    QuantumObservation,
-    env_name="quantum_circuit_env",
-    max_concurrent_envs=10,
-)
+fastapi_app = FastAPI(title="Quantum Circuit Optimizer")
 
-app.add_middleware(
+fastapi_app.add_middleware(
     CORSMiddleware,
     allow_origins=["*"],
     allow_methods=["*"],
@@ -78,7 +50,7 @@ app.add_middleware(
 )
 
 # ---------------------------------------------------------------------------
-# Gradio UI State (for interactive UI)
+# Gradio UI State
 # ---------------------------------------------------------------------------
 
 ui_env = QuantumCircuitEnvironment(seed=42)
@@ -232,37 +204,68 @@ def do_step(action_type, gate_type, qubit_0, qubit_1, parameter):
 # ---------------------------------------------------------------------------
 
 def create_gradio_app() -> gr.Blocks:
-    with gr.Blocks(title="Quantum Circuit Optimizer") as demo:
+    with gr.Blocks(title="Quantum Circuit Optimizer", theme=gr.themes.Soft()) as demo:
         gr.Markdown(
-            """# Quantum Circuit Optimizer
-            Build quantum circuits step-by-step to maximize fidelity with target states.
+            """# ⚛️ Quantum Circuit Optimizer
+            
+            Build quantum circuits step-by-step to maximize fidelity with target quantum states.
             
-            **Available Tasks:**
-            - Bell State (Easy): 2 qubits, no noise
-            - GHZ State (Medium): 3 qubits, depolarizing noise
-            - Unitary Approximation (Hard): 2 qubits, thermal noise
+            ### 🎯 Available Tasks:
+            - **Bell State (Easy)**: Create entangled 2-qubit Bell state - No noise
+            - **GHZ State (Medium)**: Create 3-qubit GHZ state - Depolarizing noise, linear connectivity
+            - **Unitary Approximation (Hard)**: Approximate complex unitary - Thermal noise, restricted connectivity
+            
+            ### 🎮 How to Use:
+            1. Select a task and click **Reset Environment**
+            2. Choose an action type (ADD gate, REMOVE last gate, SWAP qubits, tune PARAM, or STOP)
+            3. Configure gate parameters (gate type, qubits, rotation angle)
+            4. Click **Execute Step** to apply the action
+            5. Monitor fidelity, scores, and circuit depth in real-time
             """
         )
 
         with gr.Row():
-            task_dd = gr.Dropdown(choices=list(TASK_MAP.keys()), value="Bell State (Easy)", label="Task")
-            reset_btn = gr.Button("Reset Environment", variant="primary")
-
+            with gr.Column(scale=2):
+                task_dd = gr.Dropdown(
+                    choices=list(TASK_MAP.keys()), 
+                    value="Bell State (Easy)", 
+                    label="🎯 Select Task"
+                )
+            with gr.Column(scale=1):
+                reset_btn = gr.Button("🔄 Reset Environment", variant="primary", size="lg")
+
+        gr.Markdown("### 🎛️ Action Configuration")
+        
         with gr.Row():
-            action_dd = gr.Dropdown(choices=["ADD", "REMOVE", "SWAP", "PARAM", "STOP"], value="ADD", label="Action Type")
-            gate_dd = gr.Dropdown(choices=["H", "X", "CNOT", "RX", "RZ"], value="H", label="Gate (for ADD)")
+            action_dd = gr.Dropdown(
+                choices=["ADD", "REMOVE", "SWAP", "PARAM", "STOP"], 
+                value="ADD", 
+                label="Action Type"
+            )
+            gate_dd = gr.Dropdown(
+                choices=["H", "X", "CNOT", "RX", "RZ"], 
+                value="H", 
+                label="Gate (for ADD)"
+            )
 
         with gr.Row():
             qubit_0 = gr.Number(value=0, label="Qubit 0", precision=0)
             qubit_1 = gr.Number(value=1, label="Qubit 1 (CNOT/SWAP)", precision=0)
-            parameter = gr.Number(value=None, label="Parameter (RX/RZ angle)", precision=4)
+            parameter = gr.Number(value=None, label="Parameter θ (RX/RZ angle in radians)", precision=4)
 
-        step_btn = gr.Button("Execute Step", variant="secondary")
+        step_btn = gr.Button("▶️ Execute Step", variant="secondary", size="lg")
 
+        gr.Markdown("### 📊 Environment State")
+        
         status_out = gr.Textbox(label="Status", lines=3, interactive=False)
-        circuit_out = gr.Textbox(label="Circuit", lines=10, interactive=False)
-        scores_out = gr.Textbox(label="Scores", lines=6, interactive=False)
-        history_out = gr.Textbox(label="Step History", lines=8, interactive=False)
+        
+        with gr.Row():
+            with gr.Column():
+                circuit_out = gr.Textbox(label="🔗 Current Circuit", lines=12, interactive=False)
+            with gr.Column():
+                scores_out = gr.Textbox(label="📈 Scores", lines=12, interactive=False)
+        
+        history_out = gr.Textbox(label="📜 Step History", lines=10, interactive=False)
 
         OUTS = [status_out, circuit_out, scores_out, history_out]
 
@@ -277,13 +280,7 @@ def create_gradio_app() -> gr.Blocks:
 # ---------------------------------------------------------------------------
 
 demo = create_gradio_app()
-app = gr.mount_gradio_app(app, demo, path="/ui")
-
-
-@app.get("/")
-async def root():
-    """Root endpoint - redirect to Gradio UI."""
-    return RedirectResponse(url="/ui")
+app = gr.mount_gradio_app(fastapi_app, demo, path="/")
 
 
 @app.get("/health")
@@ -292,7 +289,7 @@ async def health():
     return {"status": "ok", "environment": "quantum_circuit_optimizer"}
 
 
-@app.get("/tasks")
+@app.get("/api/tasks")
 async def list_tasks():
     """List available quantum circuit optimization tasks."""
     return {