Update app.py

app.py

@@ -33,7 +33,7 @@ def process_sql_file(file_path):
                 if statement.strip():  # Skip empty statements
                     conn.execute(text(statement))
                     
-        return True, "SQL file successfully executed! Proceeding to query interface..."
+        return True, "SQL file successfully executed! Click 'Continue' to proceed to query interface..."
         
     except Exception as e:
         return False, f"Error processing SQL file: {str(e)}"
@@ -57,7 +57,7 @@ def process_csv_file(file_path):
         records = df.to_dict('records')
         insert_rows_into_table(records, table)
         
-        return True, "CSV file successfully loaded! Proceeding to query interface..."
+        return True, "CSV file successfully loaded! Click 'Continue' to proceed to query interface..."
         
     except Exception as e:
         return False, f"Error processing CSV file: {str(e)}"
@@ -176,13 +176,13 @@ def query_sql(user_query: str) -> str:
     except ValueError:
         return result
 
+# Create the Gradio interface
 with gr.Blocks() as demo:
-    # Create both interfaces at the top level
-    upload_interface = gr.Blocks()
-    query_interface = gr.Blocks(visible=False)
-
-    # Create upload interface content
-    with upload_interface:
+    current_state = gr.State("upload")
+    
+    # Upload Interface Components
+    upload_group = gr.Group(visible=True)
+    with upload_group:
         gr.Markdown("""
         # Data Query Interface
 
@@ -208,10 +208,12 @@ with gr.Blocks() as demo:
             file_types=[".csv", ".sql"],
             type="filepath"
         )
-        status = gr.Textbox(label="Status", interactive=False)
+        upload_status = gr.Textbox(label="Status", interactive=False)
+        continue_btn = gr.Button("Continue", visible=False)
 
-    # Create query interface content
-    with query_interface:
+    # Query Interface Components
+    query_group = gr.Group(visible=False)
+    with query_group:
         gr.Markdown("""
         ## Data Query Interface
         
@@ -234,47 +236,67 @@ with gr.Blocks() as demo:
         
         schema_display = gr.Markdown(value="Loading schema...")
         
-        def update_schema():
-            schema = get_table_schema()
-            if schema:
-                return f"### Current Schema:\n```\n{schema}\n```"
-            return "No data loaded"
-        
-        user_input.change(
-            fn=query_sql,
-            inputs=[user_input],
-            outputs=[query_output]
-        )
-        
         with gr.Row():
             refresh_table_btn = gr.Button("Refresh Table")
             refresh_schema_btn = gr.Button("Refresh Schema")
-            
-        refresh_table_btn.click(
-            fn=get_data_table,
-            outputs=[data_table]
-        )
-        
-        refresh_schema_btn.click(
-            fn=update_schema,
-            outputs=[schema_display]
-        )
-        
-        query_interface.load(
-            fn=update_schema,
-            outputs=[schema_display]
-        )
+            back_btn = gr.Button("Upload New File")
 
     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)
+        return message, gr.Button.update(visible=success)
+
+    def update_schema():
+        schema = get_table_schema()
+        if schema:
+            return f"### Current Schema:\n```\n{schema}\n```"
+        return "No data loaded"
 
+    def switch_to_query():
+        return {
+            upload_group: gr.Group.update(visible=False),
+            query_group: gr.Group.update(visible=True),
+            schema_display: update_schema()
+        }
+
+    def switch_to_upload():
+        return {
+            upload_group: gr.Group.update(visible=True),
+            query_group: gr.Group.update(visible=False),
+            continue_btn: gr.Button.update(visible=False),
+            upload_status: gr.Textbox.update(value="")
+        }
+
+    # Event handlers
     file_input.upload(
         fn=handle_upload,
         inputs=[file_input],
-        outputs=[status, upload_interface, query_interface]
+        outputs=[upload_status, continue_btn]
+    )
+
+    continue_btn.click(
+        fn=switch_to_query,
+        outputs=[upload_group, query_group, schema_display]
+    )
+
+    back_btn.click(
+        fn=switch_to_upload,
+        outputs=[upload_group, query_group, continue_btn, upload_status]
+    )
+
+    user_input.change(
+        fn=query_sql,
+        inputs=[user_input],
+        outputs=[query_output]
+    )
+    
+    refresh_table_btn.click(
+        fn=get_data_table,
+        outputs=[data_table]
+    )
+    
+    refresh_schema_btn.click(
+        fn=update_schema,
+        outputs=[schema_display]
     )
 
 if __name__ == "__main__":