Update app.py

app.py

@@ -22,7 +22,6 @@ def process_sql_file(file_path):
             sql_content = file.read()
             
         # Split into individual statements
-        # This assumes statements are separated by semicolons
         statements = sql_content.split(';')
         
         # Clear existing database
@@ -178,8 +177,41 @@ def query_sql(user_query: str) -> str:
         return result
 
 with gr.Blocks() as demo:
-    # First create both interfaces
-    with gr.Blocks() as query_interface:
+    # Create both interfaces at the top level
+    upload_interface = gr.Blocks()
+    query_interface = gr.Blocks(visible=False)
+
+    # Create upload interface content
+    with upload_interface:
+        gr.Markdown("""
+        # Data Query Interface
+
+        Upload your data file to begin.
+        
+        ### Supported File Types:
+        - SQL (.sql): SQL file containing CREATE TABLE and INSERT statements
+        - CSV (.csv): CSV file with headers that will be automatically converted to a table
+        
+        ### CSV Requirements:
+        - Must include headers
+        - First column will be used as the primary key
+        - Column types will be automatically detected
+        
+        ### SQL Requirements:
+        - Must contain valid SQL statements
+        - Statements must be separated by semicolons
+        - Should include CREATE TABLE and data insertion statements
+        """)
+        
+        file_input = gr.File(
+            label="Upload Data File",
+            file_types=[".csv", ".sql"],
+            type="filepath"
+        )
+        status = gr.Textbox(label="Status", interactive=False)
+
+    # Create query interface content
+    with query_interface:
         gr.Markdown("""
         ## Data Query Interface
         
@@ -232,50 +264,18 @@ with gr.Blocks() as demo:
             fn=update_schema,
             outputs=[schema_display]
         )
-    
-    # Set query interface initially invisible
-    query_interface.visible = False
-    
-    # Create upload interface
-    with gr.Blocks() as upload_interface:
-        gr.Markdown("""
-        # Data Query Interface
 
-        Upload your data file to begin.
-        
-        ### Supported File Types:
-        - SQL (.sql): SQL file containing CREATE TABLE and INSERT statements
-        - CSV (.csv): CSV file with headers that will be automatically converted to a table
-        
-        ### CSV Requirements:
-        - Must include headers
-        - First column will be used as the primary key
-        - Column types will be automatically detected
-        
-        ### SQL Requirements:
-        - Must contain valid SQL statements
-        - Statements must be separated by semicolons
-        - Should include CREATE TABLE and data insertion statements
-        """)
-        
-        file_input = gr.File(
-            label="Upload Data File",
-            file_types=[".csv", ".sql"],
-            type="filepath"
-        )
-        status = gr.Textbox(label="Status", interactive=False)
-        
-        def handle_upload(file):
-            success, message = process_uploaded_file(file)
-            if success:
-                return message, gr.Blocks.update(visible=False), gr.Blocks.update(visible=True)
-            return message, gr.Blocks.update(visible=True), gr.Blocks.update(visible=False)
-        
-        file_input.upload(
-            fn=handle_upload,
-            inputs=[file_input],
-            outputs=[status, upload_interface, query_interface]
-        )
+    def handle_upload(file):
+        success, message = process_uploaded_file(file)
+        if success:
+            return message, gr.update(visible=False), gr.update(visible=True)
+        return message, gr.update(visible=True), gr.update(visible=False)
+
+    file_input.upload(
+        fn=handle_upload,
+        inputs=[file_input],
+        outputs=[status, upload_interface, query_interface]
+    )
 
 if __name__ == "__main__":
     demo.launch(server_name="0.0.0.0", server_port=7860, share=True)