Update app.py

app.py

@@ -48,7 +48,14 @@ def get_qcs(srv, num_of_qubits, max_gates, g):
         
         st.write("Plotting...")
         fig, axs = plt.subplots(3, 2, figsize=(7,10), constrained_layout=True, dpi=120)
+
+        for ax in axs.flatten():
+            ax.axis('off')
+            ax.text(0.5, 0.5,"Circuit generated with errors")
+            
+        
         for qc,is_svr,ax in zip(qc_list, svr_list, axs.flatten()): 
+            ax.clear()
             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)
@@ -63,25 +70,20 @@ def get_qcs(srv, num_of_qubits, max_gates, g):
 # run
 
 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.subheader('Generating quantum circuits with diffusion models. Official demo of [[paper-arxiv]](https://arxiv.org/abs/2311.02041) [[code-repo]](https://github.com/FlorianFuerrutter/genQC).', divider=True)
-# st.divider()
+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).
+""")
 
 col1, col2 = st.columns(2)
 
 srv           = col1.text_input('SRV', "[1,1,1,2,2]")
 num_of_qubits = col1.radio('Number of qubits (should match SRV)', [3,4,5,6,7,8], index=2)
+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:
-    col1.warning(f'Number of qubits does not match with given SRV {srv_list}. This will result in error-circuits!', icon="⚠️")
-else:
-    col1.success(f'Number of qubits match with given SRV {srv_list}.', icon="✅")
-
-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)
+    st.warning(f'Number of qubits does not match with given SRV {srv_list}. This could result in error-circuits!', icon="⚠️")
 
 if col1.button('Generate circuits'):    
     fig = get_qcs(srv_list, num_of_qubits, max_gates, g)