import requests
import json

ocr_text = """
PT.INDOLAKTO
FAKTUR
50210604
LOWOKWARU
MALANG
TANGGAL
"""

prompt = f"""
Dari teks OCR invoice berikut:

{ocr_text}

Cari:

1. supplier
2. invoice_no
3. tanggal

Jawab HANYA dalam format JSON valid.
Jangan beri penjelasan tambahan.

Contoh:

{{
  "supplier": "PT ABC",
  "invoice_no": "123456",
  "tanggal": "2025-01-01"
}}
"""

payload = {
    "model": "gemma3:4b",
    "prompt": prompt,
    "stream": False,
    "options": {
    "temperature": 0
    }
}



response = requests.post(
    "http://localhost:11434/api/generate",
    json=payload,
    timeout=300
)

print(response.json()["response"])