Deploy text-to-SQL demo app

- Gradio interface for SQL generation
- Loads model from HuggingFace Hub
- Includes example queries

app.py

@@ -8,6 +8,7 @@ and generates PostgreSQL queries from natural language questions.
 import spaces
 import gradio as gr
 import torch
+import sqlparse
 from transformers import AutoModelForCausalLM, AutoTokenizer, BitsAndBytesConfig
 from peft import PeftModel
 
@@ -132,6 +133,14 @@ Generate only the SQL query without any explanation. [/INST]"""
         sql = sql.split("</s>")[0].strip()
         sql = sql.split("[INST]")[0].strip()
         
+        # Format SQL for better readability
+        sql = sqlparse.format(
+            sql,
+            reindent=True,
+            keyword_case='upper',
+            indent_width=2
+        )
+        
         print(f"[DEBUG] Generated SQL: {sql[:100]}...")
         return sql
         

requirements.txt

@@ -6,3 +6,4 @@ gradio==4.44.1
 torch>=2.0.0
 sentencepiece>=0.1.99
 protobuf>=3.20.0
+sqlparse>=0.5.0