Upload 4 files

API_KEY=dev

@@ -0,0 +1,5 @@
+API_KEY=dev
+OPENAI_API_KEY=sk-xxxxxxxxxxxxxxxx
+OAI_CHAT_MODEL=gpt-4o-mini
+OAI_EMB_MODEL=text-embedding-3-small
+DATABASE_URL=sqlite:///./app.db

main.py

@@ -0,0 +1,226 @@
+# main.py
+"""
+Mini Invoice/Estimate (Quote) SaaS — single-file FastAPI app
+Beginner-friendly: SQLite + SQLModel (SQLAlchemy) + Pydantic.
+
+Features (MVP)
+- Customers CRUD
+- Products CRUD
+- Quotes: create, add/remove line items, compute totals
+- Invoices: create (or convert from quote), mark paid, compute totals
+- Simple API-key auth via X-API-Key header (set API_KEY env var; defaults to "dev")
+- Autogenerated docs at /docs
+
+Run locally:
+  python -m venv .venv && source .venv/bin/activate
+  pip install -r requirements.txt
+  export API_KEY=dev
+  uvicorn main:app --reload
+"""
+
+from __future__ import annotations
+from typing import Optional, List
+from datetime import datetime, date
+import os
+
+from fastapi import FastAPI, Depends, HTTPException, Header, Query
+from fastapi.middleware.cors import CORSMiddleware
+from pydantic import BaseModel
+from sqlmodel import Field as SQLField, Session, SQLModel, create_engine, select, Relationship
+
+from openai_integration import router as ai_router
+app.include_router(ai_router, prefix="/ai", tags=["ai"])
+
+# --------------------------
+# Auth
+# --------------------------
+API_KEY = os.getenv("API_KEY", "dev")
+
+async def require_api_key(x_api_key: str | None = Header(default=None)):
+    if x_api_key != API_KEY:
+        raise HTTPException(status_code=401, detail="Invalid or missing X-API-Key")
+
+# --------------------------
+# DB
+# --------------------------
+DB_URL = os.getenv("DATABASE_URL", "sqlite:///./app.db")
+engine = create_engine(DB_URL, echo=False)
+
+def get_session():
+    with Session(engine) as session:
+        yield session
+
+# --------------------------
+# Models
+# --------------------------
+class Customer(SQLModel, table=True):
+    id: Optional[int] = SQLField(default=None, primary_key=True)
+    name: str = SQLField(index=True)
+    email: Optional[str] = None
+    phone: Optional[str] = None
+    address: Optional[str] = None
+    city: Optional[str] = None
+    country: Optional[str] = None
+
+    quotes: List["Quote"] = Relationship(back_populates="customer")
+    invoices: List["Invoice"] = Relationship(back_populates="customer")
+
+class Product(SQLModel, table=True):
+    id: Optional[int] = SQLField(default=None, primary_key=True)
+    name: str
+    unit_price: float = SQLField(ge=0)
+    currency: str = SQLField(default="JPY")
+    sku: Optional[str] = None
+    description: Optional[str] = None
+
+class Quote(SQLModel, table=True):
+    id: Optional[int] = SQLField(default=None, primary_key=True)
+    customer_id: int = SQLField(foreign_key="customer.id")
+    status: str = SQLField(default="draft", index=True)  # draft/sent/accepted/expired
+    issue_date: date = SQLField(default_factory=lambda: datetime.utcnow().date())
+    valid_until: Optional[date] = None
+    notes: Optional[str] = None
+
+    customer: Optional[Customer] = Relationship(back_populates="quotes")
+    items: List["QuoteItem"] = Relationship(back_populates="quote")
+
+class QuoteItem(SQLModel, table=True):
+    id: Optional[int] = SQLField(default=None, primary_key=True)
+    quote_id: int = SQLField(foreign_key="quote.id")
+    product_id: Optional[int] = SQLField(foreign_key="product.id", default=None)
+    description: str
+    quantity: float = SQLField(gt=0, default=1)
+    unit_price: float = SQLField(ge=0)
+    tax_rate: float = SQLField(ge=0, default=0.0)
+
+    quote: Optional[Quote] = Relationship(back_populates="items")
+
+class Invoice(SQLModel, table=True):
+    id: Optional[int] = SQLField(default=None, primary_key=True)
+    customer_id: int = SQLField(foreign_key="customer.id")
+    status: str = SQLField(default="unpaid", index=True)  # unpaid/paid/void
+    issue_date: date = SQLField(default_factory=lambda: datetime.utcnow().date())
+    due_date: Optional[date] = None
+    notes: Optional[str] = None
+
+    customer: Optional[Customer] = Relationship(back_populates="invoices")
+    items: List["InvoiceItem"] = Relationship(back_populates="invoice")
+
+class InvoiceItem(SQLModel, table=True):
+    id: Optional[int] = SQLField(default=None, primary_key=True)
+    invoice_id: int = SQLField(foreign_key="invoice.id")
+    product_id: Optional[int] = SQLField(foreign_key="product.id", default=None)
+    description: str
+    quantity: float = SQLField(gt=0, default=1)
+    unit_price: float = SQLField(ge=0)
+    tax_rate: float = SQLField(ge=0, default=0.0)
+
+    invoice: Optional[Invoice] = Relationship(back_populates="items")
+
+# --------------------------
+# Totals
+# --------------------------
+class MoneyBreakdown(BaseModel):
+    subtotal: float
+    tax: float
+    total: float
+
+def round2(v: float) -> float:
+    return float(f"{v:.2f}")
+
+def compute_totals(items: list[dict]) -> MoneyBreakdown:
+    subtotal = 0.0
+    tax = 0.0
+    for it in items:
+        line = it["quantity"] * it["unit_price"]
+        subtotal += line
+        tax += line * it.get("tax_rate", 0.0)
+    return MoneyBreakdown(subtotal=round2(subtotal), tax=round2(tax), total=round2(subtotal + tax))
+
+# --------------------------
+# App
+# --------------------------
+app = FastAPI(title="Mini Invoice/Estimate SaaS", version="0.1.0")
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"], allow_credentials=True,
+    allow_methods=["*"], allow_headers=["*"],
+)
+
+@app.on_event("startup")
+def on_startup():
+    SQLModel.metadata.create_all(engine)
+
+# -------- Customers --------
+@app.post("/customers", dependencies=[Depends(require_api_key)])
+def create_customer(payload: Customer, session: Session = Depends(get_session)):
+    session.add(payload); session.commit(); session.refresh(payload)
+    return payload
+
+@app.get("/customers", dependencies=[Depends(require_api_key)])
+def list_customers(limit: int = Query(50, ge=1, le=200), offset: int = Query(0, ge=0), session: Session = Depends(get_session)):
+    rows = session.exec(select(Customer).offset(offset).limit(limit)).all()
+    total = session.exec(select(Customer)).count()
+    return {"data": rows, "pagination": {"total": total, "limit": limit, "offset": offset}}
+
+# -------- Products --------
+@app.post("/products", dependencies=[Depends(require_api_key)])
+def create_product(payload: Product, session: Session = Depends(get_session)):
+    session.add(payload); session.commit(); session.refresh(payload)
+    return payload
+
+@app.get("/products", dependencies=[Depends(require_api_key)])
+def list_products(limit: int = Query(50, ge=1, le=200), offset: int = Query(0, ge=0), session: Session = Depends(get_session)):
+    rows = session.exec(select(Product).offset(offset).limit(limit)).all()
+    total = session.exec(select(Product)).count()
+    return {"data": rows, "pagination": {"total": total, "limit": limit, "offset": offset}}
+
+# -------- Quotes --------
+@app.post("/quotes", dependencies=[Depends(require_api_key)])
+def create_quote(payload: Quote, session: Session = Depends(get_session)):
+    if not session.get(Customer, payload.customer_id):
+        raise HTTPException(400, "Customer not found")
+    session.add(payload); session.commit(); session.refresh(payload)
+    return payload
+
+@app.get("/quotes/{quote_id}", dependencies=[Depends(require_api_key)])
+def get_quote(quote_id: int, session: Session = Depends(get_session)):
+    q = session.get(Quote, quote_id)
+    if not q: raise HTTPException(404, "Quote not found")
+    items = session.exec(select(QuoteItem).where(QuoteItem.quote_id == quote_id)).all()
+    totals = compute_totals([it.dict() for it in items])
+    return {"quote": q, "items": items, "totals": totals}
+
+@app.post("/quotes/{quote_id}/items", dependencies=[Depends(require_api_key)])
+def add_quote_item(quote_id: int, payload: QuoteItem, session: Session = Depends(get_session)):
+    if not session.get(Quote, quote_id): raise HTTPException(404, "Quote not found")
+    payload.quote_id = quote_id
+    session.add(payload); session.commit(); session.refresh(payload)
+    return payload
+
+# -------- Invoices --------
+@app.post("/invoices", dependencies=[Depends(require_api_key)])
+def create_invoice(payload: Invoice, session: Session = Depends(get_session)):
+    if not session.get(Customer, payload.customer_id):
+        raise HTTPException(400, "Customer not found")
+    session.add(payload); session.commit(); session.refresh(payload)
+    return payload
+
+@app.get("/invoices/{invoice_id}", dependencies=[Depends(require_api_key)])
+def get_invoice(invoice_id: int, session: Session = Depends(get_session)):
+    inv = session.get(Invoice, invoice_id)
+    if not inv: raise HTTPException(404, "Invoice not found")
+    items = session.exec(select(InvoiceItem).where(InvoiceItem.invoice_id == invoice_id)).all()
+    totals = compute_totals([it.dict() for it in items])
+    return {"invoice": inv, "items": items, "totals": totals}
+
+@app.post("/invoices/{invoice_id}/items", dependencies=[Depends(require_api_key)])
+def add_invoice_item(invoice_id: int, payload: InvoiceItem, session: Session = Depends(get_session)):
+    if not session.get(Invoice, invoice_id): raise HTTPException(404, "Invoice not found")
+    payload.invoice_id = invoice_id
+    session.add(payload); session.commit(); session.refresh(payload)
+    return payload
+
+@app.get("/")
+def root():
+    return {"ok": True, "app": "mini-invoice-saas", "docs": "/docs"}

openai_integration.py

@@ -0,0 +1,130 @@
+# openai_integration.py
+"""
+OpenAI (ChatGPT) integration for the Mini Invoice/Estimate SaaS (FastAPI)
+- Uses OpenAI Python SDK v1 (chat completions + embeddings)
+- Auth via env var: OPENAI_API_KEY
+"""
+
+from __future__ import annotations
+import os
+from typing import List, Optional
+
+from fastapi import APIRouter, Depends, HTTPException
+from pydantic import BaseModel, Field
+
+try:
+    from main import require_api_key  # reuse API-key header guard
+except Exception:
+    async def require_api_key():
+        return None
+
+from openai import OpenAI
+
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
+if not OPENAI_API_KEY:
+    raise RuntimeError("Set OPENAI_API_KEY before importing openai_integration")
+
+client = OpenAI(api_key=OPENAI_API_KEY)
+
+OAI_CHAT_MODEL = os.getenv("OAI_CHAT_MODEL", "gpt-4o-mini")  # 軽量モデル推奨
+OAI_EMB_MODEL = os.getenv("OAI_EMB_MODEL", "text-embedding-3-small")
+
+router = APIRouter()
+
+# -------- Schemas --------
+class LineItem(BaseModel):
+    description: str
+    quantity: float = 1
+    unit_price: float
+    tax_rate: float = 0.1
+
+class GenerateEmailRequest(BaseModel):
+    kind: str = Field(pattern="^(quote|invoice)$")
+    company_name: str
+    customer_name: str
+    language: str = Field("ja", description="ja or en")
+    items: List[LineItem]
+    due_date: Optional[str] = None
+    notes: Optional[str] = None
+    tone: str = Field("polite", description="polite|friendly|concise")
+
+class SummarizeRequest(BaseModel):
+    text: str
+    language: str = "ja"
+    max_points: int = 5
+
+class EmbeddingsRequest(BaseModel):
+    texts: List[str]
+
+class EmbeddingsResponse(BaseModel):
+    vectors: List[List[float]]
+
+# -------- Helpers --------
+EMAIL_SYS = (
+    "You are a helpful business assistant. Write concise, professional emails. "
+    "Output a subject line and a body."
+)
+SUM_SYS = (
+    "You are a world-class note taker. Produce clean bullet points and an 'Action Items' list."
+)
+
+def _chat(messages: list[dict], max_tokens: int = 600, temperature: float = 0.3) -> str:
+    resp = client.chat.completions.create(
+        model=OAI_CHAT_MODEL, messages=messages, max_tokens=max_tokens, temperature=temperature
+    )
+    return resp.choices[0].message.content.strip()
+
+def _format_items(items: List[LineItem]) -> str:
+    return "\n".join(
+        f"- {it.description}: 数量 {it.quantity}, 単価 {it.unit_price:.2f}, 税率 {it.tax_rate*100:.0f}%"
+        for it in items
+    )
+
+# -------- Routes --------
+@router.post("/generate-email", dependencies=[Depends(require_api_key)])
+async def generate_email(req: GenerateEmailRequest):
+    kind_ja = "御見積書" if req.kind == "quote" else "請求書"
+    items_block = _format_items(req.items)
+    user_prompt = f"""
+以下の情報を用いて、{kind_ja}送付メールの本文を{req.language}で作成してください。
+制約:
+- 件名(Subject)と本文を出力
+- 本文は宛名、要点の箇条書き、締め、署名の順
+- 不要な装飾は避け、{req.tone}な口調
+
+会社名: {req.company_name}
+顧客名: {req.customer_name}
+支払期日: {req.due_date or '記載なし'}
+明細:
+{items_block}
+特記事項: {req.notes or 'なし'}
+""".strip()
+
+    text = _chat(
+        [{"role": "system", "content": EMAIL_SYS}, {"role": "user", "content": user_prompt}],
+        max_tokens=500,
+    )
+    return {"email": text}
+
+@router.post("/summarize-notes", dependencies=[Depends(require_api_key)])
+async def summarize_notes(req: SummarizeRequest):
+    user_prompt = f"""
+次のメモを{req.language}で要約してください。箇条書きで最大{req.max_points}点。最後に"Action Items:"として実行項目を列挙。
+---
+{req.text}
+---
+""".strip()
+    text = _chat(
+        [{"role": "system", "content": SUM_SYS}, {"role": "user", "content": user_prompt}],
+        max_tokens=400,
+    )
+    return {"summary": text}
+
+@router.post("/embeddings", response_model=EmbeddingsResponse, dependencies=[Depends(require_api_key)])
+async def embeddings(req: EmbeddingsRequest):
+    try:
+        resp = client.embeddings.create(model=OAI_EMB_MODEL, input=req.texts)
+        vectors = [d.embedding for d in resp.data]
+        return {"vectors": vectors}
+    except Exception as e:
+        raise HTTPException(500, f"Embeddings failed: {e}")

requirements

@@ -0,0 +1,6 @@
+fastapi>=0.110
+uvicorn[standard]>=0.27
+sqlmodel>=0.0.16
+pydantic>=2.6
+python-multipart>=0.0.9
+openai>=1.40.0