Update app.py

app.py

@@ -16,18 +16,17 @@ def generate_sql(user_question, create_table_statements, instructions=""):
     
     return sql_query
 
-# Gradio 接口
-with gr.Blocks() as demo:
-    gr.Markdown("## SQL Query Generator")
-    
-    user_question = gr.Textbox(label="User Question", placeholder="请输入您的问题...", value="从纽约的客户那里获得的总收入是多少？")
-    create_table_statements = gr.Textbox(label="Create Table Statements", placeholder="请输入DDL语句...", value="CREATE TABLE customers (id INT, city VARCHAR(50), revenue DECIMAL);")
-    instructions = gr.Textbox(label="Instructions (可选)", placeholder="请输入额外说明...", value="")
-    
-    submit_btn = gr.Button("生成 SQL 查询")
-    output = gr.Textbox(label="生成的 SQL 查询")
-    
-    submit_btn.click(generate_sql, inputs=[user_question, create_table_statements, instructions], outputs=output)
+
+demo = gr.Interface(
+    fn=generate_sql,
+    inputs=[
+        gr.Markdown("## SQL Query Generator"),
+        gr.Textbox(label="User Question", placeholder="请输入您的问题...", value="从纽约的客户那里获得的总收入是多少？"),
+        gr.Textbox(label="Create Table Statements", placeholder="请输入DDL语句...", value="CREATE TABLE customers (id INT, city VARCHAR(50), revenue DECIMAL);"),
+        gr.Textbox(label="Instructions (可选)", placeholder="请输入额外说明...", value="")
+    ],
+    outputs="text",
+)
 
 
 if __name__ == "__main__":