Upload 4 files

app.py

@@ -0,0 +1,59 @@
+# Basic Workflow
+#Prompt -> LLM(Gemini Pro) -> Query -> SQL Database -> Response
+from dotenv import load_dotenv
+load_dotenv() ## load all environment variables
+import streamlit as st  
+import os
+import sqlite3
+import google.generativeai as genai
+
+##configure api key by calling from .env file
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+# Function to load Google Gemini Model and provide query as response
+def get_gemini_response(question,prompt):
+  model=genai.GenerativeModel('gemini-pro')
+  response = model.generate_content([prompt[0],question])
+  return response.text
+
+## Function to retrieve query from SQL Database. this will hit the database and get the result 
+
+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
+
+prompt = [
+  """
+  You are an expert in converting English question to SQL query! The SQL databse has the name STUDENT and has following columns - NAME, CLASS, SECTION and MARKS \n\nFor example, \nExample1 - How many entries of records are present?, the sql command will be something like this SELECT COUNT(*) FROM STUDENT;
+  \nExample 2 -Tell me all the student studying in Data Science class?, The SQL command will be something like this SELECT * FROM STUDENT where CLASS="Data Science";
+  also the sql code should not have  ``` in begining or end and sql word in the output.  
+  """
+]
+
+#Streamlit APP
+
+st.set_page_config(page_title="I can Retrieve Any SQL query")
+st.header("Gemini Powered App To Retrieve SQL Data")
+
+question = st.text_input("Input: ",key="input")
+
+submit = st.button("Ask the question")
+
+#get response
+if submit:
+  response = get_gemini_response(question,prompt)# will get SQL Query
+  print(response)
+  data = read_sql_query(response,"student.db")
+  st.subheader("The Response is: ")
+  for row in data:
+    print(row)
+    st.header(row)
+
+

requirements.txt

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

sql.py

@@ -0,0 +1,40 @@
+import sqlite3
+
+## Connecting to sqlite
+connection = sqlite3.connect("student.db")
+
+# Cursor will be responsible to insert record, create table, retrive 
+cursor = connection.cursor()
+
+##Creating Table
+table_info="""
+Create table STUDENT(NAME VARCHAR(25), CLASS VARCHAR(25), SECTION VARCHAR(25),MARKS INT);
+"""
+
+cursor.execute(table_info)
+
+## Inserting some records
+
+cursor.execute('''Insert Into STUDENT values('Atharva','AI/ML','A','95')''')
+cursor.execute('''Insert Into STUDENT values('Ashutosh','SWE','B','80')''')
+cursor.execute('''Insert Into STUDENT values('Samarth','Data Science','B','35')''')
+cursor.execute('''Insert Into STUDENT values('Yash','Data Science','A','76')''')
+cursor.execute('''Insert Into STUDENT values('Asnee','AI/ML','B','100')''')
+cursor.execute('''Insert Into STUDENT values('Sam','AI/ML','A','98')''')
+cursor.execute('''Insert Into STUDENT values('Emily','DEVOPS','A','50')''')
+cursor.execute('''Insert Into STUDENT values('Manasi','DEVOPS','B','97')''')
+
+## Display all records
+
+data = cursor.execute('''select * From STUDENT''')
+
+for row in data:
+  print(row)
+
+## Closing the connection
+  
+connection.commit()
+connection.close()
+
+
+