imkhan107 commited on
Commit
9a5faed
·
verified ·
1 Parent(s): 5028173

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -65
app.py CHANGED
@@ -1,67 +1,4 @@
1
- from dotenv import load_dotenv
2
- load_dotenv() ## load all the environemnt variables
3
-
4
  import streamlit as st
5
- import os
6
- import sqlite3
7
-
8
- import google.generativeai as genai
9
- ## Configure Genai Key
10
-
11
- genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
12
-
13
- ## Function To Load Google Gemini Model and provide queries as response
14
-
15
- def get_gemini_response(question,prompt):
16
- model=genai.GenerativeModel('gemini-pro')
17
- response=model.generate_content([prompt[0],question])
18
- return response.text
19
-
20
- ## Fucntion To retrieve query from the database
21
-
22
- def read_sql_query(sql,db):
23
- conn=sqlite3.connect(db)
24
- cur=conn.cursor()
25
- cur.execute(sql)
26
- rows=cur.fetchall()
27
- conn.commit()
28
- conn.close()
29
- for row in rows:
30
- print(row)
31
- return rows
32
-
33
- ## Define Your Prompt
34
- prompt=[
35
- """
36
- You are an expert in converting English questions to SQL query!
37
- The SQL database has the name BARADARI and has the following columns - NAME, CLASS,
38
- SECTION \n\nFor example,\nExample 1 - How many entries of records are present?,
39
- the SQL command will be something like this SELECT COUNT(*) FROM BARADARI ;
40
- \nExample 2 - Tell me all the people studying in Data Science class?,
41
- the SQL command will be something like this SELECT * FROM BARADARI
42
- where CLASS="Data Science";
43
- also the sql code should not have ``` in beginning or end and sql word in output
44
-
45
- """
46
-
47
-
48
- ]
49
-
50
- ## Streamlit App
51
-
52
- st.set_page_config(page_title="I can Retrieve Any SQL query")
53
- st.header("Gemini App To Retrieve SQL Data")
54
-
55
- question=st.text_input("Input: ",key="input")
56
-
57
- submit=st.button("Ask the question")
58
 
59
- # if submit is clicked
60
- if submit:
61
- response=get_gemini_response(question,prompt)
62
- print(response)
63
- response=read_sql_query(response,"baradari.db")
64
- st.subheader("The Response is")
65
- for row in response:
66
- print(row)
67
- st.header(row)
 
 
 
 
1
  import streamlit as st
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
2
 
3
+ x = st.slider('Select a value')
4
+ st.write(x, 'squared is', x * x)