Update app.py

app.py

@@ -1,5 +1,5 @@
 from dotenv import load_dotenv
-load_dotenv() ## load all the environemnt variables
+load_dotenv() # Load all env variables 
 
 import streamlit as st
 from PyPDF2 import PdfReader
@@ -12,7 +12,6 @@ from langchain_google_genai import ChatGoogleGenerativeAI
 from langchain.chains.question_answering import load_qa_chain
 from langchain.prompts import PromptTemplate
 from PIL import Image
-import sqlite3
 
 ## Function to load gemini pro model and get responses 
 def get_gemini_response(input, image=None):
@@ -63,20 +62,11 @@ def user_input(user_question):
     response = chain({"input_documents":docs, "question": user_question}, return_only_outputs=True)
     return response["output_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
-
 def main():
     st.set_page_config(page_title='Combined Demo')
     st.header("Combined Application")
 
-    app_mode = st.sidebar.selectbox("Choose the App Mode", ["Gemini Q&A", "PDF Q&A", "Gemini Image", "Gemini SQL"])
+    app_mode = st.sidebar.selectbox("Choose the App Mode", ["Gemini Q&A", "PDF Q&A", "Gemini Image"])
 
     if app_mode == "Gemini Q&A":
         st.subheader("Gemini LLM Application")
@@ -114,28 +104,5 @@ def main():
             st.subheader("The Response is :")
             st.write(response)
 
-    elif app_mode == "Gemini SQL":
-        st.subheader("Gemini LLM Application for SQL")
-        question = st.text_input("Input")
-        if st.button("Ask the question"):
-            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
-                """
-            ]
-            response = get_gemini_response(question, prompt)
-            st.subheader("The Response is :")
-            st.write(response)
-            st.subheader("SQL Query Result:")
-            result = read_sql_query(response, "student.db")
-            st.write(result)
-
 if __name__ == "__main__":
     main()