creat database and app

app.py

@@ -0,0 +1,71 @@
+import os
+import streamlit as st
+import sqlite3
+import google.generativeai as genai
+
+# Configure API key
+genai.configure(api_key=st.secrets["GOOGLE_API_KEY"])
+
+
+# Function to load Gemini Model and provide SQL query as response
+def get_gemini_response(prompt, question):
+    # model = genai.GenerativeModel("gemini-1.5-flash")
+    model = genai.GenerativeModel("gemini-pro")
+    response = model.generate_content([prompt[0], question])
+    return response.text
+
+
+# Function to retrieve query from the SQL database
+def read_sql_query(sql, db):
+    connection = sqlite3.connect(db)
+    cursor = connection.cursor()
+    cursor.execute(sql)
+    rows = cursor.fetchall()
+    connection.commit()
+    connection.close()
+
+    for row in rows:
+        print(row)
+    return rows
+
+
+# Define prompt
+prompt = [
+    """
+    You are an expert in converting English questions to SQL query!
+    The SQL database has 4 tables include STUDENT, LECTURER, COURSE, REGISTRATION FORM.
+    STUDENT has the following columns - ID_STU(primary key), NAME_STU, BIRTHDAY_STU and BIRTHPLACE_STU.
+    LECTURER has the following columns - ID_LEC(primary key), NAME_LEC, BIRTHDAY_LEC and BIRTHPLACE_LEC.
+    COURSE has the following columns - ID_COU(primary key), NAME_COU, ID_LEC(foreign key references LECTURER(ID_LEC)).
+    REGISTRATION_FORM has the following columns - ID_REG(primary key), ID_COU(foreign key references COURSE(ID_COU)), ID_STU(foreign key references STUDENT(ID_STU)).
+    
+    For example,
+    Example 1 - How many entries of STUDENT table are present?,
+    the SQL command will be something like this SELECT COUNT(*) FROM STUDENT;
+    Example 2 - Tell me all the students having birthplace Hải Phòng?,
+    the SQL command will be something like this SELECT * FROM STUDENT WHERE BIRTHPLACE = 'Hải Phòng';
+    Example 3 - Tell me all the course lecturer Đặng Quốc Việt teaching?,
+    the SQL command will be something like this SELECT B.NAME_COU FROM LECTURER A JOIN COURSE B ON A.ID_LEC = B.ID_LEC WHERE A.NAME_LEC = 'Đặng Quốc Việt';
+    Example 4- Tell me all the students studying 2 or more classes?,
+    the SQL command will be something like this SELECT A.NAME_STU FROM STUDENT A JOIN REGISTRATION_FORM B ON A.ID_STU = B.ID_STU GROUP BY A.NAME_STU HAVING COUNT(B.ID_COU) > 1;
+
+    note that the SQL code must not start and end with ``` and must not contain the "sql" word in output.
+"""
+]
+
+# Streamlit App
+st.set_page_config(page_title="Text to SQL")
+st.header("Gemini App to Retrieve SQL Data")
+
+question = st.text_input("Input", key="input")
+submit = st.button("Ask the question")
+
+# If submit is clicked
+if submit:
+    response = get_gemini_response(prompt, question)
+    print(response)
+
+    data = read_sql_query(response, "database/student.db")
+    st.subheader("Result")
+    for row in data:
+        st.subheader(row)

database/create_student_db.py

@@ -0,0 +1,210 @@
+import sqlite3
+
+# Connect to sqlite
+connection = sqlite3.connect("database/student.db")
+
+# Create a cursor object to create table, insert record and retrieve
+cursor = connection.cursor()
+
+# Create the table
+table_info = """ create table if not exists STUDENT (
+    ID_STU          integer primary key not null,
+    NAME_STU        nvarchar(30),
+    BIRTHDAY_STU    date,
+    BIRTHPLACE_STU  nvarchar(100))"""
+cursor.execute(table_info)
+
+table_info = """ create table if not exists LECTURER (
+    ID_LEC          integer primary key not null,
+    NAME_LEC        nvarchar(30),
+    BIRTHDAY_LEC    date,
+    BIRTHPLACE_LEC  nvarchar(100))"""
+cursor.execute(table_info)
+
+table_info = """create table if not exists COURSE (
+    ID_COU      integer primary key not null,
+    NAME_COU    nvarchar(100),
+    ID_LEC      integer,
+    foreign key (ID_LEC) references LECTURER(ID_LEC))"""
+cursor.execute(table_info)
+
+table_info = """create table if not exists REGISTRATION_FORM (
+    ID_REG integer primary key not null,
+    ID_COU integer,
+    ID_STU integer,
+    foreign key (ID_COU) references COURSE(ID_COU),
+    foreign key (ID_STU) references STUDENT(ID_STU))"""
+cursor.execute(table_info)
+
+
+# Insert some records
+connection.execute("PRAGMA foreign_keys = ON")
+
+cursor.execute(
+    """insert into STUDENT values (1, 'Đặng Tuấn Anh','1999-08-08', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into STUDENT values (2, 'Hoàng Đức Anh','1999-08-09', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into STUDENT values (3, 'Lưu Trang Anh','1999-08-10', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into STUDENT values (4, 'Phạm Hoàng Anh','2000-08-11', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into STUDENT values (5, 'Phạm Thị Hiền Anh','2000-07-12', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into STUDENT values (6, 'Phạm Khắc Việt Anh','2000-07-13', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into STUDENT values (7, 'Đỗ Hoàng Gia Bảo','2000-07-14', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into STUDENT values (8, 'Trần Thị Minh Châu','1998-07-15', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into STUDENT values (9, 'Tăng Phương Chi','1998-09-16', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into STUDENT values (10, 'Gan Feng Du','1998-09-17', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into STUDENT values (11, 'Phạm Tiến Dũng','1998-09-19', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into STUDENT values (12, 'Nguyễn Thái Dương','1997-09-19', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into STUDENT values (13, 'Trần An Dương','1997-05-20', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into STUDENT values (14, 'Mạc Trung Đức','1997-05-21', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into STUDENT values (15, 'Vũ Hương Giang','1997-05-22', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into STUDENT values (16, 'Nguyễn Thị Ngân Hà','1997-01-23', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into STUDENT values (17, 'Nguyễn Lê Hiếu','1996-01-24', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into STUDENT values (18, 'Phạm Xuân Hòa','1996-01-25', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into STUDENT values (19, 'Khoa Minh Hoàng','1996-01-26', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into STUDENT values (20, 'Nguyễn Hữu Hiệp Hoàng','1996-01-27', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into STUDENT values (21, 'Nguyễn Văn A', '2000-09-25', 'Hải Phòng')"""
+)
+cursor.execute("""insert into STUDENT (ID_STU, BIRTHPLACE_STU) values (22, 'Hà Nội')""")
+
+cursor.execute(
+    """insert into LECTURER values (1, 'Đặng Quốc Việt','1987-7-01', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into LECTURER values (2, 'Hoàng Văn Bảo','1987-7-02', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into LECTURER values (3, 'Lưu Thanh Tuấn','1987-7-03', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into LECTURER values (4, 'Hoàng Thị Thanh Mai','1987-7-04', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into LECTURER values (5, 'Nguyễn Quỳnh Hoa','1987-7-05', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into LECTURER values (6, 'Cao Thị Xuân Dung','1992-6-06', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into LECTURER values (7, 'Đỗ Hồng Việt','1992-6-07', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into LECTURER values (8, 'Phạm Thị Thu Hương','1992-6-08', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into LECTURER values (9, 'Bùi Thị Vân Thiện','1992-6-6', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into LECTURER values (10, 'Nguyễn Thị Thu Hiền','1990-5-10', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into LECTURER values (11, 'Nguyễn Thị Trà My','1990-5-11', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into LECTURER values (12, 'Trần Thị Thúy','1990-5-12', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into LECTURER values (13, 'Trần Trọng Dũng','1990-5-13', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into LECTURER values (14, 'Mạc Văn Việt','1990-1-14', 'Nam Định');"""
+)
+cursor.execute(
+    """insert into LECTURER values (15, 'Bùi Thị Thu Hương','1993-1-15', 'Hưng Yên');"""
+)
+cursor.execute(
+    """insert into LECTURER values (16, 'Nguyễn Văn Đạm','1993-1-16', 'Hải Phòng');"""
+)
+cursor.execute(
+    """insert into LECTURER values (17, 'Lê Thị Hợi','1993-1-17', 'Hà Nội');"""
+)
+cursor.execute(
+    """insert into LECTURER values (18, 'Phạm Văn Cường','1993-2-18', 'Bắc Ninh');"""
+)
+cursor.execute(
+    """insert into LECTURER values (19, 'Khoa Năng Tùng','1991-2-19', 'Nam Định');"""
+)
+
+cursor.execute("""insert into COURSE values (1,'Tester basic',1);""")
+cursor.execute("""insert into COURSE values (2,'Tester advance',2);""")
+cursor.execute("""insert into COURSE values (3,'Automation test',1);""")
+cursor.execute("""insert into COURSE values (4,'API testing',4);""")
+cursor.execute("""insert into COURSE values (5,'DB testing',5);""")
+cursor.execute("""insert into COURSE values (6,'Performance testing',3);""")
+cursor.execute("""insert into COURSE values (7,'GUI testing',7);""")
+cursor.execute("""insert into COURSE values (8,'Mobile testing',8);""")
+cursor.execute("""insert into COURSE values (9,'Game testing',9);""")
+
+cursor.execute("""insert into REGISTRATION_FORM values (1,1,1);""")
+cursor.execute("""insert into REGISTRATION_FORM values (2,2,2);""")
+cursor.execute("""insert into REGISTRATION_FORM values (3,3,3);""")
+cursor.execute("""insert into REGISTRATION_FORM values (4,4,4);""")
+cursor.execute("""insert into REGISTRATION_FORM values (5,5,5);""")
+cursor.execute("""insert into REGISTRATION_FORM values (6,6,6);""")
+cursor.execute("""insert into REGISTRATION_FORM values (7,7,7);""")
+cursor.execute("""insert into REGISTRATION_FORM values (8,8,8);""")
+cursor.execute("""insert into REGISTRATION_FORM values (9,9,9);""")
+cursor.execute("""insert into REGISTRATION_FORM values (10,1,10);""")
+cursor.execute("""insert into REGISTRATION_FORM values (11,2,11);""")
+cursor.execute("""insert into REGISTRATION_FORM values (12,3,12);""")
+cursor.execute("""insert into REGISTRATION_FORM values (13,4,13);""")
+cursor.execute("""insert into REGISTRATION_FORM values (14,5,14);""")
+cursor.execute("""insert into REGISTRATION_FORM values (15,6,15);""")
+cursor.execute("""insert into REGISTRATION_FORM values (16,7,16);""")
+cursor.execute("""insert into REGISTRATION_FORM values (17,8,17);""")
+cursor.execute("""insert into REGISTRATION_FORM values (18,9,18);""")
+cursor.execute("""insert into REGISTRATION_FORM values (19,1,19);""")
+cursor.execute("""insert into REGISTRATION_FORM values (20,2,20);""")
+cursor.execute("""insert into REGISTRATION_FORM values (21,3,1);""")
+cursor.execute("""insert into REGISTRATION_FORM values (22,4,2);""")
+cursor.execute("""insert into REGISTRATION_FORM values (23,5,3);""")
+cursor.execute("""insert into REGISTRATION_FORM values (24,6,4);""")
+cursor.execute("""insert into REGISTRATION_FORM values (25,7,5);""")
+cursor.execute("""insert into REGISTRATION_FORM values (26,8,6);""")
+cursor.execute("""insert into REGISTRATION_FORM values (27,9,7);""")
+cursor.execute("""insert into REGISTRATION_FORM values (28,1,8);""")
+cursor.execute("""insert into REGISTRATION_FORM values (29,2,9);""")
+cursor.execute("""insert into REGISTRATION_FORM values (30,3,10);""")
+
+
+# Close the connection
+connection.commit()
+connection.close()

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+google-generativeai
+# -e .

setup.py

@@ -0,0 +1,29 @@
+from setuptools import find_packages, setup
+from typing import List
+
+HYPHEN_E_DOT = "-e ."
+
+
+def get_requirements(file_path: str) -> List[str]:
+    """
+    this function will return the list of requirements
+    """
+    requirements = []
+    with open(file_path) as file_obj:
+        requirements = file_obj.readlines()
+        requirements = [req.replace("\n", "") for req in requirements]
+
+        if HYPHEN_E_DOT in requirements:
+            requirements.remove(HYPHEN_E_DOT)
+
+    return requirements
+
+
+setup(
+    name="text-to-sql",
+    version="0.0.1",
+    author="Alain De Long",
+    author_email="[email protected]",
+    packages=find_packages(),
+    install_requires=get_requirements("requirements.txt"),
+)