Upload folder using huggingface_hub

README.md

@@ -1,12 +1,34 @@
----
-title: Chichewa Text2sql
-emoji: 🔥
-colorFrom: blue
-colorTo: red
-sdk: gradio
-sdk_version: 6.6.0
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+---
+title: Chichewa Text2SQL
+emoji: 🌍
+colorFrom: blue
+colorTo: green
+sdk: gradio
+sdk_version: "4.44.0"
+app_file: app.py
+pinned: false
+hardware: zero-a10g
+license: mit
+---
+
+# Chichewa Text-to-SQL
+
+Query databases using natural language in **Chichewa** or English.
+
+Model: [johneze/Llama-3.1-8B-Instruct-chichewa-text2sql](https://huggingface.co/johneze/Llama-3.1-8B-Instruct-chichewa-text2sql)
+
+## API Usage
+
+You can call this Space programmatically via `gradio_client`:
+
+```python
+from gradio_client import Client
+
+client = Client("johneze/chichewa-text2sql")
+result = client.predict(
+    question="Ndi boma liti komwe anakolola chimanga chambiri?",
+    language="ny",
+    api_name="/generate_sql"
+)
+print(result)
+```

app.py

@@ -0,0 +1,115 @@
+"""
+Chichewa Text-to-SQL — HuggingFace Space
+Loads johneze/Llama-3.1-8B-Instruct-chichewa-text2sql and exposes a
+Gradio API endpoint that the Streamlit app (or anyone) can call.
+Uses ZeroGPU for free GPU access on HF Spaces.
+"""
+from __future__ import annotations
+
+import re
+import spaces
+import gradio as gr
+import torch
+from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline
+
+MODEL_ID = "johneze/Llama-3.1-8B-Instruct-chichewa-text2sql"
+
+tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+model = AutoModelForCausalLM.from_pretrained(
+    MODEL_ID,
+    torch_dtype=torch.bfloat16,
+    device_map="auto",
+)
+pipe = pipeline("text-generation", model=model, tokenizer=tokenizer)
+
+
+def extract_sql(text: str) -> str:
+    match = re.search(r"(?is)select\s.+", text)
+    if not match:
+        return text.strip()
+    sql = match.group(0)
+    for sep in [";", "\n"]:
+        if sep in sql:
+            sql = sql.split(sep)[0]
+    return sql.strip() + ";"
+
+
+@spaces.GPU
+def generate_sql(question: str, language: str = "ny") -> str:
+    """
+    Generate SQL from a Chichewa or English question.
+    language: 'ny' for Chichewa, 'en' for English.
+    Returns a SQL SELECT statement.
+    """
+    lang_name = "Chichewa" if language == "ny" else "English"
+
+    messages = [
+        {
+            "role": "system",
+            "content": (
+                "You are an expert Text-to-SQL model for a SQLite database "
+                "with the following tables: production, population, food_insecurity, "
+                "commodity_prices, mse_daily. "
+                "Given a natural language question, generate ONE valid SQL SELECT query. "
+                "Return ONLY the SQL query, no explanation."
+            ),
+        },
+        {
+            "role": "user",
+            "content": f"Language: {lang_name}\nQuestion: {question}",
+        },
+    ]
+
+    prompt = tokenizer.apply_chat_template(
+        messages, tokenize=False, add_generation_prompt=True
+    )
+
+    out = pipe(
+        prompt,
+        max_new_tokens=128,
+        do_sample=False,
+        pad_token_id=tokenizer.eos_token_id,
+    )[0]["generated_text"]
+
+    generated = out[len(prompt):] if out.startswith(prompt) else out
+    return extract_sql(generated)
+
+
+# ── Gradio UI ──────────────────────────────────────────────────────────────
+with gr.Blocks(title="Chichewa Text-to-SQL") as demo:
+    gr.Markdown("# 🌍 Chichewa Text-to-SQL\nEnter a question in Chichewa or English to generate SQL.")
+
+    with gr.Row():
+        question_box = gr.Textbox(
+            label="Question",
+            placeholder="Ndi boma liti komwe anakolola chimanga chambiri?",
+            lines=3,
+        )
+        language_box = gr.Radio(
+            ["ny", "en"],
+            value="ny",
+            label="Language",
+        )
+
+    submit_btn = gr.Button("Generate SQL", variant="primary")
+    sql_output = gr.Code(label="Generated SQL", language="sql")
+
+    submit_btn.click(
+        fn=generate_sql,
+        inputs=[question_box, language_box],
+        outputs=sql_output,
+    )
+
+    gr.Examples(
+        examples=[
+            ["Ndi boma liti komwe anakolola chimanga chambiri?", "ny"],
+            ["Which district produced the most Maize?", "en"],
+            ["Ndi anthu angati ku Lilongwe?", "ny"],
+            ["What is the food insecurity level in Nsanje?", "en"],
+        ],
+        inputs=[question_box, language_box],
+    )
+
+
+if __name__ == "__main__":
+    demo.launch()

requirements.txt

@@ -0,0 +1,6 @@
+gradio>=4.44.0
+transformers>=4.44.0
+torch>=2.1.0
+accelerate>=0.34.0
+safetensors>=0.4.0
+spaces>=0.19.0