Upload app.py with huggingface_hub

app.py

@@ -0,0 +1,66 @@
+import os
+import gradio as gr
+from huggingface_hub import InferenceClient
+
+HF_TOKEN = os.environ.get("HF_TOKEN")
+client = InferenceClient(
+    model="Qwen/Qwen2.5-Coder-32B-Instruct",
+    token=HF_TOKEN,
+)
+
+SYSTEM_PROMPT = """You are a SQL expert. Given a database schema and a question in English, generate a DuckDB-compatible SQL query that answers the question.
+
+Rules:
+- Return ONLY the SQL query, no explanation, no markdown, no code fences
+- Use exact table and column names from the schema
+- Use DuckDB SQL syntax
+- Add LIMIT 100 unless the user asks for a specific count or all rows
+- Keep queries simple: use the fewest tables and JOINs possible
+- For text search, always use ILIKE '%term%' (case-insensitive) instead of exact match
+- Never use LOWER() or UPPER() for comparison, use ILIKE instead
+- If the question is not related to querying the database (e.g. personal questions, general knowledge, chitchat), respond with exactly: NOT_A_DATA_QUESTION
+- Only generate SQL for questions that can be answered by querying the provided database schema"""
+
+
+def generate_sql(question: str, schema_ddl: str) -> str:
+    prompt = f"Database schema:\n{schema_ddl}\n\nQuestion: {question}\n\nSQL:"
+
+    sql = ""
+    for token in client.chat_completion(
+        messages=[
+            {"role": "system", "content": SYSTEM_PROMPT},
+            {"role": "user", "content": prompt},
+        ],
+        max_tokens=500,
+        temperature=0.1,
+        stream=True,
+    ):
+        chunk = token.choices[0].delta.content or ""
+        sql += chunk
+        yield sql.strip()
+
+    # Clean up final result
+    sql = sql.strip()
+    if sql.startswith("```"):
+        sql = sql.split("\n", 1)[1] if "\n" in sql else sql[3:]
+    if sql.endswith("```"):
+        sql = sql[:-3]
+    sql = sql.strip()
+    if sql.lower().startswith("sql"):
+        sql = sql[3:].strip()
+
+    yield sql
+
+
+demo = gr.Interface(
+    fn=generate_sql,
+    inputs=[
+        gr.Textbox(label="Question", placeholder="Show me all patients who died during their hospital stay"),
+        gr.Textbox(label="Schema DDL", lines=10, placeholder="CREATE TABLE patients (...)"),
+    ],
+    outputs=gr.Textbox(label="Generated SQL"),
+    title="TinyEHR Text-to-SQL",
+    description="Generate SQL queries for the TinyEHR dataset from natural language.",
+)
+
+demo.launch()