feat: add main menu options and scripts

app.py

@@ -1,5 +1,6 @@
 import os
 import sys
+import time
 
 # insert current directory to sys.path
 sys.path.insert(0, os.path.abspath(os.path.dirname(__file__)))
@@ -25,7 +26,7 @@ st.set_page_config(
 
 # TEXT_2_SQL_API = "http://83.219.197.235:40172/api/text2sql/ask"
 TEXT_2_SQL_API = os.environ.get(
-    "TEXT_2_SQL_API", "http://localhost:8501/api/text2sql/ask"
+    "TEXT_2_SQL_API", "http://213.181.122.2:40057/api/text2sql/ask"
 )
 
 
@@ -49,14 +50,15 @@ def execute_sql(sql_query):
         return cursor.fetchall()
     except Exception as e:
         st.info("Database is not supported")
+        return None
 
 
-@st.cache_data
+# @st.cache_data
 def ask_text2sql(question, context):
     if detect(question) != "en":
         question = translate_question(question)
-        st.write("The question is translated to Vietnamese:")
-        st.code(question, language="en")
+        # st.write("The question is translated to Vietnamese:")
+        # st.code(question, language="en")
 
     r = requests.post(
         TEXT_2_SQL_API,
@@ -182,11 +184,26 @@ def examples():
 
             if example_btns[idx]:
                 st.markdown("##### SQL query:")
-                query = ask_text2sql(row["question"], row["context"])
-                st.code(format_sql(query), language="sql")
-                result = execute_sql(query)
-                st.write("The result of the SQL query is:")
-                st.dataframe(pd.DataFrame(result))
+                tries = 3
+                with st.spinner("Generating 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
 
 
 # Define a function for the Stock Prediction page
@@ -218,13 +235,26 @@ CREATE TABLE product (id number, category text, name text, description text, pri
 
     if get_sql_button:
         st.markdown("##### Output")
-        query = ask_text2sql(question, context)
-        st.write("The SQL query generated by the model is:")
-        # Display the SQL query in a code block
-        st.code(format_sql(query), language="sql")
-        result = execute_sql(query)
-        st.write("The result of the SQL query is:")
-        st.dataframe(pd.DataFrame(result))
+        tries = 3
+        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
 
 
 # Define a function for the About page

resources/examples.csv

@@ -1,5 +1,9 @@
 context,question
-"CREATE TABLE head (name VARCHAR, born_state VARCHAR, age VARCHAR)","List the name, born state and age of the heads of departments ordered by age."
+"CREATE TABLE customer (id number, name text, gender text, age number, district_id number;
+CREATE TABLE registration (customer_id number, product_id number); 
+CREATE TABLE district (id number, name text, prefix text, province_id number); 
+CREATE TABLE province (id number, name text, code text)
+CREATE TABLE product (id number, category text, name text, description text, price number, duration number, data_amount number, voice_amount number, sms_amount number);",Số lượng khách hàng có độ tuổi từ 15 đến 20 tuổi
 "CREATE TABLE customer (id number, name text, gender text, age number, district_id number;
 CREATE TABLE registration (customer_id number, product_id number); 
 CREATE TABLE district (id number, name text, prefix text, province_id number); 
@@ -9,4 +13,4 @@ CREATE TABLE product (id number, category text, name text, description text, pri
 CREATE TABLE registration (customer_id number, product_id number); 
 CREATE TABLE district (id number, name text, prefix text, province_id number); 
 CREATE TABLE province (id number, name text, code text)
-CREATE TABLE product (id number, category text, name text, description text, price number, duration number, data_amount number, voice_amount number, sms_amount number);",Số lượng khách hàng có độ tuổi từ 15 đến 20 tuổi
+CREATE TABLE product (id number, category text, name text, description text, price number, duration number, data_amount number, voice_amount number, sms_amount number);", Liệt kê số khách hàng của từng tỉnh