first commit

Author: rahul-aot

.gitignore

@@ -0,0 +1,22 @@
+# Python
+__pycache__/
+*.py[cod]
+*$py.class
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Distribution / packaging
+dist/
+build/
+*.egg-info/
+
+# IDE
+.vscode/
+.idea/

backend/.env.example

@@ -0,0 +1,11 @@
+ENV=development
+
+# Gemini
+GEMINI_API_KEY=your_gemini_api_key_here
+
+# PostgreSQL (READ-ONLY USER)
+DB_HOST=localhost
+DB_PORT=5432
+DB_NAME=your_db_name
+DB_USER=readonly_user
+DB_PASSWORD=readonly_password

backend/Dockerfile

@@ -0,0 +1,24 @@
+FROM python:3.11-slim
+
+WORKDIR /app
+
+# Install system dependencies (needed for psycopg2)
+RUN apt-get update && apt-get install -y \
+    gcc \
+    libpq-dev \
+    && rm -rf /var/lib/apt/lists/*
+
+# Copy requirements
+COPY requirements.txt .
+
+# Install python dependencies
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy application code
+COPY . .
+
+# Expose port
+EXPOSE 8000
+
+# Command to run the application
+CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]

backend/README.md

@@ -0,0 +1,36 @@
+from fastapi import APIRouter, HTTPException
+
+from app.schemas.request import SQLQueryRequest
+from app.schemas.response import SQLQueryResponse
+from app.services.sql_generator import generate_sql
+from app.services.sql_executor import execute_sql
+
+router = APIRouter(tags=["Query"])
+
+
+@router.post("/query", response_model=SQLQueryResponse)
+def query_sql(payload: SQLQueryRequest):
+    try:
+        # Step 1: Generate SQL + explanation using Gemini
+        result = generate_sql(payload.question)
+
+        sql = result["sql"]
+        explanation = result["explanation"]
+
+        # Step 2: Execute SQL safely
+        columns, rows = execute_sql(sql)
+
+        return SQLQueryResponse(
+            sql=sql,
+            explanation=explanation,
+            columns=columns,
+            rows=rows,
+        )
+
+    except ValueError as exc:
+        raise HTTPException(status_code=400, detail=str(exc))
+    except Exception:
+        raise HTTPException(
+            status_code=500,
+            detail="Failed to process SQL query"
+        )

backend/app/api/__init__.py

@@ -0,0 +1,26 @@
+from pydantic_settings import BaseSettings
+
+
+class Settings(BaseSettings):
+    # App
+    APP_NAME: str = "AskSQLAI"
+    ENV: str = "development"
+
+    # Gemini
+    GEMINI_API_KEY: str
+    GEMINI_MODEL: str = "gemini-1.5-flash"
+
+    # Database (read-only)
+    DB_HOST: str
+    DB_PORT: int = 5432
+    DB_NAME: str
+    DB_USER: str
+    DB_PASSWORD: str
+
+    class Config:
+        env_file = ".env"
+        env_file_encoding = "utf-8"
+        extra = "ignore"
+
+
+settings = Settings()

backend/app/api/query.py

@@ -0,0 +1,9 @@
+from pydantic import BaseModel, Field
+
+
+class SQLQueryRequest(BaseModel):
+    question: str = Field(
+        ...,
+        min_length=5,
+        description="Natural language question to be converted into SQL"
+    )