Update app.py

app.py

@@ -38,15 +38,21 @@ pipeline = load_pipeline()
 is_gpu_busy = False
 def get_qcs(srv, num_of_qubits, max_gates, g):
     global is_gpu_busy
-       
-    out_tensor           = generate_srv_tensors(pipeline, f"Generate SRV: {srv}", samples=6, system_size=num_of_qubits, num_of_qubits=num_of_qubits, max_gates=max_gates, g=g) 
-    qc_list, _, svr_list = convert_tensors_to_srvs(out_tensor, pipeline.gate_pool)
-
-    fig, axs = plt.subplots(3, 2, figsize=(7,10), constrained_layout=True, dpi=120)
-    for qc,is_svr,ax in zip(qc_list, svr_list, axs.flatten()): 
-        qc.draw("mpl", plot_barriers=False, ax=ax)
-        ax.set_title(f"{'Correct' if is_svr==srv else 'NOT correct'}, is SRV = {is_svr}")
 
+    with st.status("Generation started", expanded=True) as status:
+        st.write("Generating tensors...")
+        out_tensor = generate_srv_tensors(pipeline, f"Generate SRV: {srv}", samples=6, system_size=num_of_qubits, num_of_qubits=num_of_qubits, max_gates=max_gates, g=g)
+        
+        st.write("Converting to circuits...")
+        qc_list, _, svr_list = convert_tensors_to_srvs(out_tensor, pipeline.gate_pool)
+        
+        st.write("Plotting...")
+        fig, axs = plt.subplots(3, 2, figsize=(7,10), constrained_layout=True, dpi=120)
+        for qc,is_svr,ax in zip(qc_list, svr_list, axs.flatten()): 
+            qc.draw("mpl", plot_barriers=False, ax=ax)
+            ax.set_title(f"{'Correct' if is_svr==srv else 'NOT correct'}, is SRV = {is_svr}")
+        status.update(label="Generation complete!", state="complete", expanded=False)
+    
     # buf = io.BytesIO()
     # fig.savefig(buf)
     # buf.seek(0)
@@ -60,6 +66,7 @@ st.title("genQC · Generative Quantum Circuits")
 st.write("""
 Generating quantum circuits with diffusion models. Official demo of [[paper-arxiv]](https://arxiv.org/abs/2311.02041) [[code-repo]](https://github.com/FlorianFuerrutter/genQC).
 """)
+st.divider()
 
 col1, col2 = st.columns(2)
 
@@ -68,7 +75,12 @@ num_of_qubits = col1.radio('Number of qubits (should match SRV)', [3,4,5,6,7,8],
 max_gates     = col1.select_slider('Max gates', options=[4,8,12,16,20,24,28], value=16)
 g             = col1.slider('Guidance scale', min_value=0.0, max_value=15.0, value=7.5)
 
+srv_list = ast.literal_eval(srv)
+
+if len(srv_list)!=num_of_qubits:
+    st.warning('Number of qubits does not match with given SRV. This will result in error-circuits!', icon="⚠️")
+
 if col1.button('Generate circuits'):    
-    fig = get_qcs(ast.literal_eval(srv), num_of_qubits, max_gates, g)
+    fig = get_qcs(srv_list, num_of_qubits, max_gates, g)
     # col2.image(image, use_column_width=True)
     col2.pyplot(fig)