feat: add main menu options and scripts

app.py

@@ -173,6 +173,22 @@ def examples():
     example_tabs = st.tabs([f"Example {i+1}" for i in range(len(example_df))])
     example_btns = []
 
+    with st.sidebar:
+        # create a blank space
+        st.write("")
+        st.write("")
+        st.write("")
+        execute_sql_query = st.checkbox(
+            "Execute SQL query",
+        )
+        num_tries = st.number_input(
+            "Number of tries",
+            value=3,
+            min_value=1,
+            max_value=10,
+            step=1,
+        )
+
     for idx, row in example_df.iterrows():
         with example_tabs[idx]:
             st.markdown("##### Context:")
@@ -184,26 +200,39 @@ def examples():
 
             if example_btns[idx]:
                 st.markdown("##### SQL query:")
-                tries = 3
+                tries = num_tries
                 with st.spinner("Generating SQL query..."):
-                    while tries > 0:
+                    if execute_sql_query:
+                        while tries > 0:
+                            start_time = time.time()
+                            query = ask_text2sql(row["question"], row["context"])
+                            end_time = time.time()
+                            st.write(
+                                "The SQL query generated by the model in **{:.2f}s** is:".format(
+                                    end_time - start_time
+                                )
+                            )
+                            st.code(format_sql(query), language="sql")
+                            result = execute_sql(query)
+
+                            st.write(
+                                "Executing the SQL query yields the following result:"
+                            )
+                            st.dataframe(pd.DataFrame(result), hide_index=True)
+                            if result is not None:
+                                break
+                            else:
+                                tries -= 1
+                    else:
                         start_time = time.time()
                         query = ask_text2sql(row["question"], row["context"])
                         end_time = time.time()
-                        st.write(
-                            "The SQL query generated by the model in {:.2f}s is:".format(
+                        st.markdown(
+                            "The SQL query generated by the model in **{:.2f}s** is:".format(
                                 end_time - start_time
                             )
                         )
                         st.code(format_sql(query), language="sql")
-                        result = execute_sql(query)
-
-                        st.write("Executing the SQL query yields the following result:")
-                        st.dataframe(pd.DataFrame(result), hide_index=True)
-                        if result is not None:
-                            break
-                        else:
-                            tries -= 1
 
 
 # Define a function for the Stock Prediction page
@@ -233,28 +262,57 @@ CREATE TABLE product (id number, category text, name text, description text, pri
     )
     get_sql_button = st.button("Generate SQL query")
 
+    with st.sidebar:
+        # create a blank space
+        st.write("")
+        st.write("")
+        st.write("")
+        execute_sql_query = st.checkbox(
+            "Execute SQL query",
+        )
+        num_tries = st.number_input(
+            "Number of tries",
+            value=3,
+            min_value=1,
+            max_value=10,
+            step=1,
+        )
+
     if get_sql_button:
         st.markdown("##### Output")
-        tries = 3
-        while tries > 0:
+        tries = num_tries
+        if execute_sql_query:
+            while tries > 0:
+                start_time = time.time()
+                query = ask_text2sql(question, context)
+                end_time = time.time()
+
+                st.write(
+                    "The SQL query generated by the model in **{:.2f}s** is:".format(
+                        end_time - start_time
+                    )
+                )
+                # Display the SQL query in a code block
+                st.code(format_sql(query), language="sql")
+                result = execute_sql(query)
+                st.write("Executing the SQL query yields the following result:")
+                st.dataframe(pd.DataFrame(result), hide_index=True)
+                if result is not None:
+                    break
+                else:
+                    tries -= 1
+        else:
             start_time = time.time()
             query = ask_text2sql(question, context)
             end_time = time.time()
 
-            st.write(
-                "The SQL query generated by the model in {:.2f}s is:".format(
+            st.markdown(
+                "The SQL query generated by the model in **{:.2f}s** is:".format(
                     end_time - start_time
                 )
             )
             # Display the SQL query in a code block
             st.code(format_sql(query), language="sql")
-            result = execute_sql(query)
-            st.write("Executing the SQL query yields the following result:")
-            st.dataframe(pd.DataFrame(result), hide_index=True)
-            if result is not None:
-                break
-            else:
-                tries -= 1
 
 
 # Define a function for the About page
@@ -330,7 +388,7 @@ def references():
 st.sidebar.title("Menu")
 pages = [
     "Introduction",
-    "Datasets",
+    # "Datasets",
     "Examples",
     "Interactive Demo",
     "About",