Update app.py

app.py

@@ -7,39 +7,37 @@ from IPython.display import IFrame
 
 # query params exist
 try:
-    options = ['cat', 'dog', 'mouse', 'bat', 'duck']
-
+    options = ['Book', 'Grok', 'Playground', 'Circuit']
     query_params = st.experimental_get_query_params()
     query_option = query_params['option'][0] #throws an exception when visiting http://host:port
-
-    option_selected = st.sidebar.selectbox('Pick option',
-                                            options,
-                                            index=options.index(query_option))
+    option_selected = st.sidebar.selectbox('Pick option', options, index=options.index(query_option))
     if option_selected:
         st.experimental_set_query_params(option=option_selected)
+        SetIframeURL(option_selected)
 
 # run when query params don't exist. e.g on first launch
 except: # catch exception and set query param to predefined value
-    options = ['cat', 'dog', 'mouse', 'bat', 'duck']
+    options = ['Book', 'Grok', 'Playground', 'Circuit']
     st.experimental_set_query_params(option=options[1]) # defaults to dog
-
     query_params = st.experimental_get_query_params()
     query_option = query_params['option'][0]
-
-    option_selected = st.sidebar.selectbox('Pick option',
-                                            options,
-                                            index=options.index(query_option))
+    option_selected = st.sidebar.selectbox('Pick option', options, index=options.index(query_option))
     if option_selected:
         st.experimental_set_query_params(option=option_selected)
-
-        
-src1='https://oreilly-qc.github.io?p=2-1'
-src2='https://javafxpert.github.io/grok-bloch/'
-src3='https://davidbkemp.github.io/quantum-gate-playground/'
-src4='https://algassert.com/quirk#circuit={%22cols%22:[[%22H%22],[%22Bloch%22],[%22Measure%22]]}'
-
-st.components.v1.iframe(src1, width=1200, height=800, scrolling=False)
-
+        SetIframeURL(option_selected)
+
+def SetIframeURL(option_selected):
+    if (option_selected=='Book'):
+        src='https://oreilly-qc.github.io?p=2-1'
+    if (option_selected=='Grok'):
+        src='https://javafxpert.github.io/grok-bloch/'
+    if (option_selected=='Playground'):
+        src='https://davidbkemp.github.io/quantum-gate-playground/'
+    if (option_selected=='Circuit'):
+        src='https://algassert.com/quirk#circuit={%22cols%22:[[%22H%22],[%22Bloch%22],[%22Measure%22]]}'
+
+# Render iframe contents
+st.components.v1.iframe(src, width=1200, height=800, scrolling=False)
 
 title = "AI Quantum - QGAN"
 description = "Using Superposition Advantage from Quantum for QGAN AI."