text-to-SQL

.gitignore

@@ -0,0 +1 @@
+.env

README.md

@@ -0,0 +1,10 @@
+---
+title: Text To SQL LLM App
+emoji: 👁
+colorFrom: indigo
+colorTo: pink
+sdk: streamlit
+sdk_version: 1.39.0
+app_file: app.py
+pinned: false
+---

app.py

@@ -0,0 +1,71 @@
+import google.generativeai as genai #type:ignore
+import streamlit as st
+# importing for env variables loader
+from dotenv import load_dotenv
+import sqlite3
+import os
+
+load_dotenv()
+
+# configure our api key
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+
+# function to load Model and provide sql query as response
+def get_gemini_response(prompt, question):
+    model = genai.GenerativeModel("gemini-1.5-flash")
+    response = model.generate_content([prompt[0], question])
+    return response.text
+
+
+# function to retrieve query from the sql data base
+def read_sql_query(sql, db):
+    conn = sqlite3.connect(db)
+    cur = conn.cursor()
+    cur.execute(sql)
+    rows = cur.fetchall()
+    conn.commit()
+    conn.close()
+    for row in rows:
+        print(row)
+    return rows
+
+
+# definning your prompt
+
+prompt = [
+    """
+    You are an expert in converting English questions to SQL queries! 
+    The SQL database has the name STUDENT and has the following columns - NAME, CLASS, 
+    SECTION, and MARKS. 
+
+    For Example:
+    - Example 1: How many entries of records are present? 
+      The SQL command will be: SELECT COUNT(*) FROM STUDENT;
+    - Example 2: Tell me all the students studying in Data Science class? 
+      The SQL command will be: SELECT * FROM STUDENT WHERE CLASS="Data Science";
+
+    Provide the clean query that can be executed on the database without using any backticks or additional characters.
+    The SQL command should not include any formatting characters or the word "SQL".
+    """
+]
+
+# also the SQL code should not have ''' in biggining or end and sql word in output.
+
+# streamlit app    
+st.set_page_config(page_title="retrieve sql queryF")
+st.header(" Gemini App to Retrieve SQL Data")
+
+user_question = st.text_input("input:", key="input")
+
+sumbit = st.button("ask the question")
+
+if sumbit:
+    sql_query = get_gemini_response(prompt,user_question)
+    print(sql_query)
+    data = read_sql_query(sql_query,"student.db")
+    st.subheader("The resplonse is:")
+    for row in data:
+        print(row)
+        st.write(row)
+        

pyproject.toml

@@ -0,0 +1,17 @@
+[tool.poetry]
+name = "text to sql llm app"
+version = "0.1.0"
+description = ""
+authors = ["Nasir Abbas <akhtarabbas.islamicreaction@gmail.com>"]
+readme = "README.md"
+
+[tool.poetry.dependencies]
+python = "^3.12"
+streamlit = "^1.39.0"
+google-generativeai = "^0.8.3"
+python-dotenv = "^1.0.1"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"

requirements.txt

@@ -0,0 +1,3 @@
+streamlit
+google-generativeai
+python-dotenv

sql.py

@@ -0,0 +1,59 @@
+import sqlite3
+
+## conect the sql
+connection = sqlite3.connect("student.db")
+
+
+# Create a cursor object to insert record, craete table, retreeve 
+cursor =connection.cursor()
+
+## create the table 
+
+table_info = """
+Create table STUDENT(NAME VARCHAR(25), CLASS VARCHAR(25),
+SECTION VARCHAR(25), MARKS INT);
+
+"""
+
+
+cursor.execute(table_info)
+
+
+# inser some more recordes
+
+cursor.execute('''Insert into STUDENT values('Nasir','data Science', 'A', 90)''')
+cursor.execute('''Insert into STUDENT values('Akhtar','Generative AI', 'B', 87)''')
+cursor.execute('''Insert into STUDENT values('Asad','DEVOPS', 'A', 83)''')
+cursor.execute('''Insert into STUDENT values('Munawer','data Science', 'C', 93)''')
+cursor.execute('''Insert into STUDENT values('Muzafar','Generative AI', 'A', 83)''')
+cursor.execute('''Insert into STUDENT values('Ali','DEVOPS', 'C', 82)''')
+cursor.execute('''Insert into STUDENT values('Kashif','data Science', 'A', 79)''')
+cursor.execute('''Insert into STUDENT values('Aneela','DEVOPS', 'B', 81)''')
+cursor.execute('''Insert into STUDENT values('Danish','data Science', 'A', 78)''')
+cursor.execute('''Insert into STUDENT values('Rabia','DEVOPS', 'C', 82)''')
+cursor.execute('''Insert into STUDENT values('Khizar','Generative AI', 'A', 76)''')
+cursor.execute('''Insert into STUDENT values('Nazia','data Science', 'A', 73)''')#
+cursor.execute('''Insert into STUDENT values('Bilal','Generative AI', 'B', 72)''')
+cursor.execute('''Insert into STUDENT values('sajid','DEVOPS', 'A', 87)''')
+cursor.execute('''Insert into STUDENT values('Asif','data Science', 'C', 63)''')
+cursor.execute('''Insert into STUDENT values('Husnain','Generative AI', 'A', 98)''')
+cursor.execute('''Insert into STUDENT values('Hamza','DEVOPS', 'C', 84)''')
+cursor.execute('''Insert into STUDENT values('Ikram','data Science', 'A', 65)''')
+cursor.execute('''Insert into STUDENT values('Ramzan','DEVOPS', 'B', 78)''')
+cursor.execute('''Insert into STUDENT values('Abid','data Science', 'A', 72)''')
+cursor.execute('''Insert into STUDENT values('Jafir','DEVOPS', 'C', 81)''')
+cursor.execute('''Insert into STUDENT values('Sadiq','Generative AI', 'A', 92)''')
+
+
+## Display all the recards
+print("The inserted recards are")
+
+data = cursor.execute('''Select * From STUDENT''')
+
+for row in data:
+    print(row)
+    
+    
+# close the connection 
+connection.commit()
+connection.close()