Update app.py

app.py

@@ -6,7 +6,7 @@ import google.generativeai as genai
 import time
 
 # Initialize Gemini
-genai.configure(api_key="AIzaSyCnuJMrTkckScv0YxAaaQzH1LDomNUUppA")
+genai.configure(api_key="YOUR_GEMINI_API_KEY")
 genai_model = genai.GenerativeModel('gemini-pro')
 
 class SQLPromptModel:
@@ -89,14 +89,27 @@ with gr.Blocks(title="House Database Query") as demo:
     gr.Markdown("""### The database contains information about different properties including their fundamental details. 
                 You can query this database using natural language.""")
     
-    # Query input and output
-    query_input = gr.Textbox(
-        lines=2,
-        label="Database Query",
-        placeholder="Enter your query or choose from examples below. Default: 'Properties in India'"
-    )
+    with gr.Row():
+        # Query input and output
+        query_input = gr.Textbox(
+            lines=2,
+            label="Database Query",
+            placeholder="Enter your query or choose from examples below. Default: 'Properties in India'"
+        )
+    
+    with gr.Row():
+        # Add submit button
+        submit_btn = gr.Button("Submit Query", variant="primary")
+    
+    with gr.Row():
+        query_output = gr.JSON(label="Query Results")
     
-    query_output = gr.JSON(label="Query Results")
+    # Connect submit button to the query function
+    submit_btn.click(
+        fn=execute_sql_query,
+        inputs=query_input,
+        outputs=query_output
+    )
     
     # Example queries
     gr.Examples(
@@ -109,9 +122,6 @@ with gr.Blocks(title="House Database Query") as demo:
         outputs=query_output,
         fn=execute_sql_query
     )
-    
-    # Submit button
-    query_input.submit(fn=execute_sql_query, inputs=query_input, outputs=query_output)
 
 if __name__ == "__main__":
     demo.launch(share=True)