|
from dotenv import load_dotenv
|
|
load_dotenv()
|
|
|
|
import streamlit as st
|
|
import os
|
|
import sqlite3
|
|
|
|
import google.generativeai as genai
|
|
|
|
|
|
genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
|
|
|
|
|
|
|
|
def get_gemini_response(question,prompt):
|
|
model=genai.GenerativeModel('gemini-pro')
|
|
response=model.generate_content([prompt[0],question])
|
|
return response.text
|
|
|
|
|
|
|
|
def read_sql_query(sql,db):
|
|
conn=sqlite3.connect(db)
|
|
cur=conn.cursor()
|
|
cur.execute(sql)
|
|
rows=cur.fetchall()
|
|
conn.commit()
|
|
conn.close()
|
|
|
|
|
|
return rows
|
|
|
|
|
|
prompt=[
|
|
"""
|
|
You are an expert in converting English questions to SQL query!
|
|
The SQL database has the name COMPANY and has the following columns - FIRST_NAME, LAST_NAME,DEPT,
|
|
SALARY,AGE \n\nFor example,\nExample 1 - How many entries of records are present?,
|
|
the SQL command will be something like this SELECT COUNT(*) FROM COMPANY ;
|
|
\nExample 2 - Tell me all the students studying in Data Science class?,
|
|
the SQL command will be something like this SELECT * FROM company
|
|
where CLASS="Data Science";
|
|
also the sql code should not have ``` in beginning or end and sql word in output
|
|
|
|
"""
|
|
|
|
|
|
]
|
|
|
|
|
|
|
|
st.set_page_config(page_title="Retrieve Any SQL query")
|
|
st.header("App To Generate SQL & Data")
|
|
|
|
question=st.text_input("Input: ",key="input")
|
|
|
|
submit=st.button("Ask the question")
|
|
|
|
|
|
if submit:
|
|
response=get_gemini_response(question,prompt)
|
|
|
|
st.subheader("Generated SQL Query")
|
|
st.code(response, language='sql')
|
|
data=read_sql_query(response,"company.db")
|
|
st.subheader("The Response is")
|
|
for row in data:
|
|
|
|
st.write(row)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|