Update app.py

app.py

@@ -27,11 +27,11 @@ class SQLPromptModel:
         schema = cursor.fetchall()
         if schema:
             return schema 
-        else:
+        else
             return None
         #return schema if schema else None
 
-    def text2sql_gemini(self, schema, user_prompt, inp_prompt=None):
+    def text2sql_gemini(self, schema, def_query, user_query=None):
         # Convert table columns to string format
         table_columns = ', '.join([f"{col[1]} {col[2]}" for col in schema])
         #col[1] - column name/index 
@@ -40,13 +40,13 @@ class SQLPromptModel:
         # Create prompt for Gemini AI
         prompt = f"""Below are SQL table schemas paired with instructions that describe a task.
         Using valid SQLite, write a response that appropriately completes the request for the provided tables.
-        ### Instruction: {user_prompt} ### 
+        ### Instruction: {def_query} ### 
         Input: CREATE TABLE sql_pdf({table_columns});
-        ### Response: (Return only generated query based on user_prompt , nothing extra)"""
+        ### Response: (Return only generated query based on def_query , nothing extra)"""
 
         # Replace default prompt with user input if provided
-        if inp_prompt is not None:
-            prompt = prompt.replace(user_prompt, inp_prompt + " ")
+        if user_query is not None:
+            prompt = prompt.replace(def_query, user_query + " ")
             
         # Get SQL query from Gemini
         completion = genai_model.generate_content(prompt)
@@ -72,13 +72,13 @@ class SQLPromptModel:
         self.conn.commit()
         return rows, columns
 
-def execute_sql_query(input_prompt):
+def execute_sql_query(user_query):
     # Database file path
     database = r"sql_pdf.db"
     sql_model = SQLPromptModel(database)
     
     # Default prompt if none provided
-    user_prompt = "Give complete details of properties in India"
+    def_query = "Give complete details of properties in India"
     
     # Try operation up to 3 times
     for _ in range(3):
@@ -87,10 +87,10 @@ def execute_sql_query(input_prompt):
             table_schema = sql_model.fetch_table_schema("sql_pdf")
             if table_schema:
                 # Generate and execute query
-                if input_prompt.strip():
-                    query = sql_model.text2sql_gemini(table_schema, user_prompt, input_prompt)
+                if user_query.strip():
+                    query = sql_model.text2sql_gemini(table_schema, def_query, user_query)
                 else:
-                    query = sql_model.text2sql_gemini(table_schema, user_prompt, user_prompt)
+                    query = sql_model.text2sql_gemini(table_schema, def_query, def_query)
                     
                 rows, columns = sql_model.execute_query(query)
                 # Return formatted results