Update app.py

app.py

@@ -1,9 +1,7 @@
-import os, re, json, time, sqlite3, hashlib, secrets, random, sys
+import os, re, json, time, sqlite3, hashlib, secrets, random
 from datetime import datetime, timedelta
 from pathlib import Path
 import requests
-
-# ── Třetí strany ──
 import gradio as gr
 
 try:
@@ -41,7 +39,6 @@ PRICE_IDS = {
 if STRIPE_OK and STRIPE_SECRET_KEY:
     _stripe.api_key = STRIPE_SECRET_KEY
 
-# Používáme Mistral přes bezplatné HF Inference API
 HF_MODEL = "mistralai/Mistral-7B-Instruct-v0.3"
 HF_API   = "https://router.huggingface.co/hf-inference/models/" + HF_MODEL
 HF_HDR   = {"Authorization": f"Bearer {HF_TOKEN}"} if HF_TOKEN else {}
@@ -184,12 +181,13 @@ def ai_extract(text: str):
     )
 
     try:
+        # Použijeme text-generation endpoint přes HF Router
         r = requests.post(
             HF_API, headers=HF_HDR,
             json={"inputs": prompt, "parameters": {"max_new_tokens": 512, "temperature": 0.05, "return_full_text": False, "do_sample": False}},
             timeout=45
         )
-        if r.status_code == 503:
+        if r.status_code == 503: # Model se nabootovává
             time.sleep(5)
             r = requests.post(HF_API, headers=HF_HDR, json={"inputs": prompt, "parameters": {"max_new_tokens": 512, "return_full_text": False}}, timeout=45)
         if r.status_code != 200:
@@ -211,7 +209,6 @@ def _parse_json(text: str):
         except: pass
     return None
 
-# (Regex fallback a Demo data zůstávají nezměněny pro plynulost kódu)
 def _parse_amount(s: str):
     try:
         s = re.sub(r'[^\d,.]', '', s).replace(",", ".")
@@ -266,6 +263,16 @@ def make_demo(filename: str, user_id: int) -> dict:
 # ═════════════════════════════════════════════════════════════════════════════
 # PARSE PIPELINE
 # ═════════════════════════════════════════════════════════════════════════════
+def check_limit(user_id, plan):
+    month = datetime.now().strftime("%Y-%m")
+    conn = db()
+    n = conn.execute("SELECT COUNT(*) FROM invoices WHERE user_id=? AND strftime('%Y-%m',created_at)=?", (user_id, month)).fetchone()[0]
+    conn.close()
+    lim = PLANS[plan]["limit"]
+    if lim == -1: return {"ok": True, "used": n, "limit": "∞"}
+    rem = lim - n
+    return {"ok": rem > 0, "used": n, "limit": lim, "msg": f"Monthly limit of {lim} invoices reached." if rem <= 0 else ""}
+
 def parse_and_save(file_path: str, filename: str, user_id: int, plan: str) -> dict:
     ext = Path(filename).suffix.lower()
     text = ocr_file(file_path, ext)
@@ -356,16 +363,6 @@ def fn_upload(files, token):
     lim_after = check_limit(user["id"], user["plan"])
     return f"✅ Processed **{len(results)}** file(s). Usage: {_usage_bar(lim_after['used'], lim_after['limit'])}", "\n\n---\n\n".join(results), json.dumps(last_data, ensure_ascii=False, indent=2, default=str)
 
-def check_limit(user_id, plan):
-    month = datetime.now().strftime("%Y-%m")
-    conn = db()
-    n = conn.execute("SELECT COUNT(*) FROM invoices WHERE user_id=? AND strftime('%Y-%m',created_at)=?", (user_id, month)).fetchone()[0]
-    conn.close()
-    lim = PLANS[plan]["limit"]
-    if lim == -1: return {"ok": True, "used": n, "limit": "∞"}
-    rem = lim - n
-    return {"ok": rem > 0, "used": n, "limit": lim, "msg": f"Monthly limit of {lim} invoices reached." if rem <= 0 else ""}
-
 def fn_list(token):
     user = get_user(token)
     if not user: return "❌ Not logged in."