Upload 6 files

Dockerfile

@@ -0,0 +1,51 @@
+# Use Ubuntu 18.04 as the base image
+FROM ubuntu:18.04
+
+# Set environment variables for locale
+ENV LANG=C.UTF-8
+ENV LC_ALL=C.UTF-8
+
+# Install system dependencies and add Python 3.10 repository
+RUN apt-get update && \
+    apt-get install -y \
+    software-properties-common && \
+    add-apt-repository ppa:deadsnakes/ppa && \
+    apt-get update && \
+    apt-get install -y \
+    python3.10 \
+    python3.10-distutils \
+    python3-pip \
+    build-essential \
+    libblas-dev \
+    liblapack-dev \
+    libbz2-dev \
+    zlib1g-dev \
+    liblzma-dev \
+    libsnappy-dev && \
+    rm -rf /var/lib/apt/lists/*
+
+# Upgrade pip
+RUN python3.10 -m pip install --upgrade pip
+
+# Install Python packages
+RUN python3.10 -m pip install \
+    streamlit \
+    google-generativeai \
+    langchain \
+    PyPDF2 \
+    chromadb \
+    faiss-cpu \
+    pdf2image
+
+# Copy application files
+COPY sql.py /app/sql.py
+COPY sqlite.py /app/sqlite.py
+
+# Set the working directory
+WORKDIR /app
+
+# Set Python 3.10 as the default Python
+RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python3.10 1
+
+# Start the application
+CMD ["streamlit", "run", "sql.py"]

requirements.txt

@@ -0,0 +1,7 @@
+streamlit
+google-generativeai
+langchain
+PyPDF2
+chromadb
+faiss-cpu
+pdf2image

sql.py

@@ -0,0 +1,75 @@
+# from dotenv import load_dotenv
+# load_dotenv() ## load all the environemnt variables
+
+import streamlit as st
+import os
+import sqlite3
+
+import google.generativeai as genai
+import os
+
+# Configure GenAI Key directly using environment variable or fallback to a hardcoded key
+API_KEY = os.getenv("GOOGLE_API_KEY", "AIzaSyCOAMr9QxGUHwUO70hyHQSghHxFgtNyPsg")
+genai.configure(api_key=API_KEY)
+
+# Function To Load Google Gemini Model and provide queries as response
+def get_gemini_response(question, prompt):
+    response = genai.generate_content([prompt, question])
+    return response.text
+
+
+## Function To Load Google Gemini Model and provide queries as response
+
+def get_gemini_response(question,prompt):
+    model=genai.GenerativeModel('gemini-pro')
+    response=model.generate_content([prompt[0],question])
+    return response.text
+
+## Fucntion To retrieve query from the database
+
+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
+
+## Define Your Prompt
+prompt=[
+    """
+    You are an expert in converting English questions to SQL query!
+    The SQL database has the name STUDENT and has the following columns - NAME, CLASS, 
+    SECTION \n\nFor example,\nExample 1 - 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 students 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 beginning or end and sql word in output
+
+    """
+
+
+]
+
+## Streamlit App
+
+st.set_page_config(page_title="I can Retrieve Any SQL query")
+st.header("SQL Data Retrieve")
+
+question=st.text_input("Input: ",key="input")
+
+submit=st.button("Ask the question")
+
+# if submit is clicked
+if submit:
+    response=get_gemini_response(question,prompt)
+    print(response)
+    response=read_sql_query(response,"student.db")
+    st.subheader("The Response is")
+    for row in response:
+        print(row)
+        st.header(row)

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,36 @@
+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 isnerted records are")
+data=cursor.execute('''Select * from STUDENT''')
+for row in data:
+    print(row)
+
+## Commit your changes int he databse
+connection.commit()
+connection.close()
+