Update app.py

app.py

@@ -8,27 +8,7 @@ import os
 
 genai.configure(api_key=os.getenv("GOOGLE_API_KEY"))
 model=genai.GenerativeModel('gemini-pro')
-
-
-#dun to retrieve query from the sql database
-
-def read_sql_query(sql,db):
-    conn=sqlite3.connect(db)
-    cursor=conn.cursor()
-    cursor.execute(sql)
-    rows=cursor.fetchall()
-    conn.commit()
-    conn.close()
-
-    for row in rows:
-        print(row)
-    return rows
-
-# prompt="""
-# you are a professinal sql query expert, i will provide input what i want from database 
-# ,you give me the query without quotes are comma or " "  double quotes or ```, just give me query
-# """
-
+ 
 
 prompt=[
     """
@@ -49,23 +29,34 @@ prompt=[
     also the sql code should not have ``` in beginning or end and sql word in output and 
      
     """
-
-
 ]
 
- 
 
+#llm response
 def gemini_sql_query(prompt,input):
     response=model.generate_content([prompt[0],input])
     return response.text
+    
+#dun to retrieve query from the sql database
+def read_sql_query(sql,db):
+    conn=sqlite3.connect(db)
+    cursor=conn.cursor()
+    cursor.execute(sql)
+    rows=cursor.fetchall()
+    conn.commit()
+    conn.close()
+    for row in rows:
+        print(row)
+    return rows
 
 
 st.set_page_config("DataChat: Explore Your Database")
 st.header("chat with your sql database")
 
-input=st.text_input("enter your input/question ")
+input=st.text_input("enter your input/question and specify correct table name")
 
-#uploaded file
+
+#save uploaded file 
 def save_uploaded_file(uploaded_file):
     file_path = os.path.join(os.getcwd(), "uploaded.db")
     with open(file_path, "wb") as f:
@@ -104,13 +95,4 @@ if submit and uploaded_file and input:
         if st.button("Copy SQL Query"):
             st.write("SQL query copied to clipboard!")
             st.text_area("SQL Query:", value=query, height=100)
-
-# if submit:
-#     query=gemini_sql_query(prompt,input)
-#     response=read_sql_query(query,"student.db")
-#     print(query)
-#     st.header("response")
-#     for row in response:
-#         values = [str(value) for value in row]
-#         st.write(*values)
-
+