Upload 7 files

.env

@@ -0,0 +1 @@
+GOOGLE_API_KEY = "AIzaSyCuh8F92Mm3UCoFLg74tugLQdPOo_yA4i0"

requirements.txt

@@ -0,0 +1,9 @@
+streamlit
+google-generativeai
+python-dotenv
+langchain
+PyPDF2
+chromadb
+faiss-cpu
+pdf2image
+sqlite3

sql.py

@@ -0,0 +1,124 @@
+
+
+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 Google Gemini API Key
+genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
+
+
+## Function To Load Google Gemini Model and provide queries as response
+def get_gemini_response(question, prompt):
+    model = genai.GenerativeModel('gemini-2.0-flash')
+    response = model.generate_content([prompt[0], question])
+    return response.text
+
+
+## Function To Retrieve Query from the Database
+def read_sql_query(sql, db):
+    try:
+        conn = sqlite3.connect(db)
+        cur = conn.cursor()
+        cur.execute(sql)
+        rows = cur.fetchall()
+        conn.commit()
+        conn.close()
+        return rows
+    except sqlite3.Error as e:
+        return [("Error:", str(e))]
+
+
+## Ensure Table Exists Before Querying
+def ensure_table_exists():
+    conn = sqlite3.connect("student.db")
+    cursor = conn.cursor()
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS STUDENT(
+        NAME VARCHAR(25),
+        CLASS VARCHAR(25),
+        SECTION VARCHAR(25),
+        MARKS INT);
+    """)
+    conn.commit()
+    conn.close()
+
+
+## Define 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, MARKS.\n\nFor example:
+    - How many entries of records are present?
+      SQL: SELECT COUNT(*) FROM STUDENT;
+    - Tell me all the students studying in Data Science class?
+      SQL: SELECT * FROM STUDENT WHERE CLASS="Data Science";
+
+    **Do NOT include "```sql" or "```" in the response.**
+    """
+]
+
+## 🎨 Streamlit App UI & UX Enhancements
+st.set_page_config(page_title="Advanced SQL Query Generator", layout="wide")
+
+# 🎯 Sidebar for Instructions & About
+with st.sidebar:
+    st.image("https://cdn-icons-png.flaticon.com/512/2721/2721292.png", width=100)
+    st.title("📌 How It Works")
+    st.markdown(
+        "1️⃣ **Enter a question in plain English**  \n"
+        "2️⃣ **The AI converts it into an SQL query**  \n"
+        "3️⃣ **The app fetches results from the database**  \n"
+        "4️⃣ **You get an answer instantly! 🎯**"
+    )
+
+    st.warning("⚠️ **First, run all three files (sqlite.py, sqlite_copy.py, and sql.py) to ensure the database exists.** Then, ask your query.")
+    st.markdown("---")
+    st.write("👩‍💻 **Created By:** Prarthana 💙")
+
+# 🎬 Animated Header
+st.markdown(
+    "<h1 style='text-align: center; color: #1f77b4;'>🎯 AI-Powered SQL Query Generator 🚀</h1>",
+    unsafe_allow_html=True
+)
+
+# 📝 User Input
+question = st.text_input("🔍 **Enter Your Question:**", key="input")
+
+# 🎯 Button to Generate SQL Query
+if st.button("🔎 Get SQL Query & Fetch Data"):
+    if question:
+        ensure_table_exists()  # Ensure table exists before querying
+
+        with st.spinner("🤖 Generating SQL query..."):
+            sql_query = get_gemini_response(question, prompt)
+
+        st.success(f"✅ Generated Query: `{sql_query}`")
+
+        # Fetch Data
+        with st.spinner("📡 Fetching Data from Database..."):
+            response = read_sql_query(sql_query, "student.db")
+
+        if response:
+            st.subheader("📊 **Query Results:**")
+            for row in response:
+                st.write(row)
+        else:
+            st.warning("⚠️ No data found in the database.")
+    else:
+        st.error("❌ Please enter a valid question.")
+
+# ✨ Footer
+st.markdown("<br><br>", unsafe_allow_html=True)
+st.markdown("<h4 style='text-align: center;'>✨ Powered by Google Gemini Pro ✨</h4>", unsafe_allow_html=True)
+
+
+
+
+

sqlite copy.py

@@ -0,0 +1,32 @@
+# Import module
+import sqlite3
+
+# Connecting to sqlite
+conn = sqlite3.connect('test.db')
+
+# Creating a cursor object using the
+# cursor() method
+cursor = conn.cursor()
+
+# Creating table
+table = """CREATE TABLE STUDENT(NAME VARCHAR(255), CLASS VARCHAR(255), 
+SECTION VARCHAR(255));"""
+cursor.execute(table)
+
+# Queries to INSERT records.
+cursor.execute('''INSERT INTO STUDENT VALUES ('Krish', 'Data Science', 'A')''')
+cursor.execute('''INSERT INTO STUDENT VALUES ('Darius', 'Data Science', 'B')''')
+cursor.execute('''INSERT INTO STUDENT VALUES ('Sudhanshu', 'Devops', 'C')''')
+cursor.execute('''INSERT INTO STUDENT VALUES ('Vikash', 'Data Science', 'C')''')
+
+# Display data inserted
+print("Data Inserted in the table: ")
+data = cursor.execute('''SELECT * FROM STUDENT''')
+for row in data:
+    print(row)
+
+# Commit your changes in the database
+conn.commit()
+
+# Closing the connection
+conn.close()

sqlite.py

@@ -0,0 +1,35 @@
+import sqlite3
+
+## Connectt to SQlite
+connection=sqlite3.connect("student.db")
+
+# Create a cursor object to insert record,create table
+
+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)
+
+## Insert Some more records
+
+cursor.execute('''Insert Into STUDENT values('Krish','Data Science','A',90)''')
+cursor.execute('''Insert Into STUDENT values('Sudhanshu','Data Science','B',100)''')
+cursor.execute('''Insert Into STUDENT values('Darius','Data Science','A',86)''')
+cursor.execute('''Insert Into STUDENT values('Vikash','DEVOPS','A',50)''')
+cursor.execute('''Insert Into STUDENT values('Dipesh','DEVOPS','A',35)''')
+
+## Disspaly ALl the records
+
+print("The inserted records are")
+data=cursor.execute('''Select * from STUDENT''')
+for row in data:
+    print(row)
+
+## Commit your changes int he databse
+connection.commit()
+connection.close()