Update app.py

app.py

@@ -196,11 +196,28 @@ with gr.Blocks(theme=theme, css=CSS, title="Quread.ai — State Vector Studio")
                 gr.Markdown("<div class='section-title'>Gate Palette</div>")
                 with gr.Row():
                     gate_H = gr.Button("H")
+                    gate_T = gr.Button("T")
+                    gate_Tdg = gr.Button("T†")
                     gate_X = gr.Button("X")
+                with gr.Row():
                     gate_Y = gr.Button("Y")
                     gate_Z = gr.Button("Z")
+                    gate_sx = gr.Button("√X")
+                    gate_sz = gr.Button("√Z")
+                with gr.Row():
+                    gate_rx_pi = gr.Button("Rx(π)")
+                    gate_rx_pi2 = gr.Button("Rx(π/2)")
+                    gate_ry_pi = gr.Button("Ry(π)")
+                    gate_ry_pi2 = gr.Button("Ry(π/2)")
+                with gr.Row():
+                    gate_rz_pi = gr.Button("Rz(π)")
+                    gate_rz_pi2 = gr.Button("Rz(π/2)")
+                    gate_I = gr.Button("I")
+                    gate_Idg = gr.Button("I†")  # will map to I
+                with gr.Row():
                     gate_S = gr.Button("S")
-                    gate_T = gr.Button("T")
+                    gate_Sdg = gr.Button("S†")
+
 
                 target = gr.Dropdown(choices=[0, 1], value=0, label="Target qubit")
 
@@ -283,11 +300,29 @@ with gr.Blocks(theme=theme, css=CSS, title="Quread.ai — State Vector Studio")
     )
 
     gate_H.click(fn=lambda: set_gate("H"), outputs=[selected_gate_state, status])
+    gate_T.click(fn=lambda: set_gate("T"), outputs=[selected_gate_state, status])
+    gate_Tdg.click(fn=lambda: set_gate("T†"), outputs=[selected_gate_state, status])
+    
     gate_X.click(fn=lambda: set_gate("X"), outputs=[selected_gate_state, status])
     gate_Y.click(fn=lambda: set_gate("Y"), outputs=[selected_gate_state, status])
     gate_Z.click(fn=lambda: set_gate("Z"), outputs=[selected_gate_state, status])
+    
+    gate_sx.click(fn=lambda: set_gate("√X"), outputs=[selected_gate_state, status])
+    gate_sz.click(fn=lambda: set_gate("√Z"), outputs=[selected_gate_state, status])
+    
+    gate_rx_pi.click(fn=lambda: set_gate("RX(π)"), outputs=[selected_gate_state, status])
+    gate_rx_pi2.click(fn=lambda: set_gate("RX(π/2)"), outputs=[selected_gate_state, status])
+    gate_ry_pi.click(fn=lambda: set_gate("RY(π)"), outputs=[selected_gate_state, status])
+    gate_ry_pi2.click(fn=lambda: set_gate("RY(π/2)"), outputs=[selected_gate_state, status])
+    gate_rz_pi.click(fn=lambda: set_gate("RZ(π)"), outputs=[selected_gate_state, status])
+    gate_rz_pi2.click(fn=lambda: set_gate("RZ(π/2)"), outputs=[selected_gate_state, status])
+    
+    gate_I.click(fn=lambda: set_gate("I"), outputs=[selected_gate_state, status])
+    gate_Idg.click(fn=lambda: set_gate("I"), outputs=[selected_gate_state, status])  # treat I† as I
+    
     gate_S.click(fn=lambda: set_gate("S"), outputs=[selected_gate_state, status])
-    gate_T.click(fn=lambda: set_gate("T"), outputs=[selected_gate_state, status])
+    gate_Sdg.click(fn=lambda: set_gate("S†"), outputs=[selected_gate_state, status])
+
 
     apply_gate_btn.click(
         fn=apply_selected_gate,