Update app.py

app.py

@@ -13,7 +13,7 @@ import time
 import sqlite3
 import hashlib
 
-print("Initializing OmniParse AI Enterprise Terminal v5.0 with User Accounts...")
+print("Initializing OmniParse AI Enterprise Terminal v5.5 with Optional Free Plan...")
 
 # ─── DATABASE INITIALIZATION ────────────────────────────────────────────────
 DB_FILE = "omniparse.db"
@@ -59,6 +59,9 @@ def db_register(username, password):
         return json.dumps({"status": "ERROR", "message": "Zadej jméno i heslo, nesmí to bejt prázdný!"})
     
     username = username.strip().lower()
+    if username == "guest":
+        return json.dumps({"status": "ERROR", "message": "Jméno 'guest' je rezervované pro Free Plan!"})
+        
     try:
         conn = sqlite3.connect(DB_FILE)
         cursor = conn.cursor()
@@ -92,7 +95,7 @@ def db_login(username, password):
         
         user_id, stored_hash, salt = row
         
-        # Podpora pro session refresh (pokud uživatel reloadne stránku a má token v sessionStorage)
+        # Podpora pro session refresh
         if password == "SESSION_REFRESH":
             cursor.execute("SELECT COUNT(*), SUM(fields_count) FROM invoices WHERE user_id = ?", (user_id,))
             stats = cursor.fetchone()
@@ -215,9 +218,11 @@ def parse_invoice(image_b64: str, username: str):
 
         invoice_fp = extracted_fields.get("INVOICE_NUMBER", extracted_fields.get("TOTAL", "UNKNOWN"))
         
-        # Kontrola duplicity v rámci přihlášeného uživatele
+        # Kontrola duplicity v rámci přihlášeného uživatele (pokud to není host)
         is_duplicate = False
-        if username and invoice_fp != "UNKNOWN":
+        is_guest = (not username or username.strip().lower() == "guest")
+
+        if not is_guest and invoice_fp != "UNKNOWN":
             try:
                 conn = sqlite3.connect(DB_FILE)
                 cursor = conn.cursor()
@@ -253,10 +258,9 @@ def parse_invoice(image_b64: str, username: str):
             "extracted_data": extracted_fields,
         }
 
-        # Zápis do DB pro konkrétního přihlášeného usera
-        fields_count = len(extracted_fields)
-        user_docs, user_fields = 1, fields_count
-        if username:
+        # Zápis do DB pouze pokud uživatel NENÍ host
+        if not is_guest:
+            fields_count = len(extracted_fields)
             try:
                 conn = sqlite3.connect(DB_FILE)
                 cursor = conn.cursor()
@@ -270,19 +274,21 @@ def parse_invoice(image_b64: str, username: str):
                     )
                     conn.commit()
                     
-                    # Vytáhnout aktualizované kumulativní statistiky
                     cursor.execute("SELECT COUNT(*), SUM(fields_count) FROM invoices WHERE user_id = ?", (user_id,))
                     count_row = cursor.fetchone()
-                    user_docs = count_row[0] or 0
-                    user_fields = count_row[1] or 0
+                    result["user_stats"] = {
+                        "docs_parsed": count_row[0] or 0,
+                        "fields_extracted": count_row[1] or 0
+                    }
                 conn.close()
             except Exception as db_err:
                 print(f"DB Insert failure: {db_err}")
-
-        result["user_stats"] = {
-            "docs_parsed": user_docs,
-            "fields_extracted": user_fields
-        }
+        else:
+            # Pro hosta vrátíme flag, JS si spočítá lokální session statistiky sám
+            result["user_stats"] = {
+                "docs_parsed": "GUEST_MODE",
+                "fields_extracted": "GUEST_MODE"
+            }
 
         # Generování CSV exportu
         rows = [["Field", "Value"]]
@@ -425,8 +431,9 @@ body{font-family:'Inter',sans-serif;background:var(--bg);color:var(--text);}
 
 <div id="toast-root" style="position:fixed;top:16px;right:16px;z-index:9999;display:flex;flex-direction:column;gap:8px;"></div>
 
+<!-- ═══ 1. AUTHENTICATION OVERLAY SCREEN ═══ -->
 <div id="auth-screen" class="fixed inset-0 z-50 flex items-center justify-center bg-[#09090b] p-4">
-  <div class="card w-full max-w-md p-8 space-y-6">
+  <div class="card w-full max-w-md p-8 space-y-5">
     <div class="text-center space-y-2">
       <div class="mx-auto w-12 h-12 rounded-xl bg-gradient-to-br from-blue-500 to-blue-700 flex items-center justify-center shadow-lg shadow-blue-500/25">
         <svg width="22" height="22" viewBox="0 0 16 16" fill="none"><path d="M2 3h12M2 6h8M2 9h10M2 12h6" stroke="white" stroke-width="1.5" stroke-linecap="round"/><circle cx="13" cy="11" r="2.5" stroke="white" stroke-width="1.2"/><path d="M15 13l1.5 1.5" stroke="white" stroke-width="1.2" stroke-linecap="round"/></svg>
@@ -435,29 +442,40 @@ body{font-family:'Inter',sans-serif;background:var(--bg);color:var(--text);}
       <p class="text-xs text-zinc-400">Multi-User Document Data-Mining Terminal</p>
     </div>
     
-    <div class="space-y-4">
+    <div class="space-y-3">
       <div>
-        <label class="block text-xs font-semibold uppercase tracking-wider text-zinc-500 mb-1">Uživatelské jméno</label>
-        <input type="text" id="auth-username" class="chat-input" placeholder="Zadej login...">
+        <label class="block text-[10px] font-bold uppercase tracking-wider text-zinc-500 mb-1">Uživatelské jméno</label>
+        <input type="text" id="auth-username" class="chat-input text-xs" placeholder="Zadej login...">
       </div>
       <div>
-        <label class="block text-xs font-semibold uppercase tracking-wider text-zinc-500 mb-1">Heslo</label>
-        <input type="password" id="auth-password" class="chat-input" placeholder="••••••••">
+        <label class="block text-[10px] font-bold uppercase tracking-wider text-zinc-500 mb-1">Heslo</label>
+        <input type="password" id="auth-password" class="chat-input text-xs" placeholder="••••••••">
       </div>
     </div>
     
-    <div class="flex flex-col gap-2 pt-2">
-      <button onclick="handleAuthAction('login')" class="btn-primary w-full py-3 font-semibold text-sm">
+    <div class="flex flex-col gap-2 pt-1">
+      <button onclick="handleAuthAction('login')" class="btn-primary w-full py-2.5 font-semibold text-xs">
         Přihlásit se do systému ⚡
       </button>
-      <button onclick="handleAuthAction('register')" class="btn-secondary w-full py-3 text-sm">
-        Registrovat nový účet
+      <button onclick="handleAuthAction('register')" class="btn-secondary w-full py-2.5 text-xs">
+        Registrovat nový cloudový účet
+      </button>
+      
+      <div class="relative flex py-2 items-center">
+        <div class="flex-grow border-t border-zinc-800"></div>
+        <span class="flex-shrink mx-3 text-[10px] text-zinc-600 font-bold uppercase">Nebo</span>
+        <div class="flex-grow border-t border-zinc-800"></div>
+      </div>
+      
+      <button onclick="bypassAuthAsGuest()" class="w-full py-2 bg-zinc-900 border border-zinc-800 hover:border-zinc-700 text-zinc-300 rounded-xl text-xs font-semibold transition-all">
+        Pokračovat bez registrace (Free Plan) →
       </button>
     </div>
   </div>
 </div>
 
 
+<!-- ═══ 2. MAIN APPLICATION WORKSPACE ═══ -->
 <div id="app-screen" style="display: none;">
   <header style="border-bottom:1px solid var(--border);background:rgba(9,9,11,.92);backdrop-filter:blur(16px); position:sticky; top:0; z-index:40;">
     <div style="max-width:1280px;margin:0 auto;padding:0 24px;height:56px;display:flex;align-items:center;justify-content:space-between;">
@@ -473,17 +491,18 @@ body{font-family:'Inter',sans-serif;background:var(--bg);color:var(--text);}
           <span id="status-dot" style="width:8px;height:8px;border-radius:50%;background:#34d399;"></span>
           <span id="status-text" style="font-size:12px;color:var(--text2);font-weight:500;">System Ready</span>
         </div>
-        <span id="user-display" class="text-sm font-medium text-blue-400"></span>
-        <button onclick="handleLogout()" class="text-xs text-red-400 hover:underline">Odhlásit se</button>
+        <span id="user-display" class="text-sm font-medium"></span>
+        <button onclick="handleLogout()" class="text-xs text-red-400 hover:underline">Zpět na login / Odhlásit</button>
       </div>
     </div>
   </header>
   
   <main style="max-width:1280px;margin:0 auto;padding:24px;">
+    <!-- Stats Banner -->
     <div class="grid grid-cols-4 gap-3 mb-5">
       <div class="card p-4 flex items-center gap-3">
         <div class="text-2xl">📄</div>
-        <div><p class="text-[10px] text-zinc-500 uppercase font-bold">Moje Dokumenty</p><p id="st-docs" class="text-lg font-bold text-white">0</p></div>
+        <div><p class="text-[10px] text-zinc-500 uppercase font-bold">Zpracované dokumenty</p><p id="st-docs" class="text-lg font-bold text-white">0</p></div>
       </div>
       <div class="card p-4 flex items-center gap-3">
         <div class="text-2xl">⚡</div>
@@ -499,7 +518,9 @@ body{font-family:'Inter',sans-serif;background:var(--bg);color:var(--text);}
       </div>
     </div>
     
+    <!-- Workspace Grid -->
     <div style="display:grid;grid-template-columns:1fr 1fr;gap:20px;">
+      <!-- Left Controls -->
       <div class="space-y-4">
         <div class="card p-5 space-y-4">
           <div class="flex justify-between items-center">
@@ -544,6 +565,7 @@ body{font-family:'Inter',sans-serif;background:var(--bg);color:var(--text);}
         </div>
       </div>
       
+      <!-- Right Terminal & Terminal Assist -->
       <div class="space-y-4">
         <div class="card p-5 space-y-3">
           <div class="flex justify-between items-center"><span class="text-sm font-semibold">Live Extraction Terminal</span><span class="section-label">Output JSON</span></div>
@@ -580,6 +602,7 @@ body{font-family:'Inter',sans-serif;background:var(--bg);color:var(--text);}
   </main>
 </div>
 
+<!-- ═══ 3. GRADIO PIPELINE JAVASCRIPT BRIDGE ═══ -->
 <script>
 let currentBase64 = "";
 let latestExtractedJson = "";
@@ -588,6 +611,10 @@ let authPolling = null;
 let parsePolling = null;
 let chatPolling = null;
 
+// Lokální paměť statistik pro uživatele na Free Planu (neukládá se do DB)
+let guestDocsCount = 0;
+let guestFieldsCount = 0;
+
 function showToast(msg, type='info') {
     const root = document.getElementById("toast-root");
     const toast = document.createElement("div");
@@ -605,26 +632,41 @@ function checkLoginState() {
     if (user) {
         document.getElementById("auth-screen").style.display = "none";
         document.getElementById("app-screen").style.display = "block";
-        document.getElementById("user-display").innerText = `👤 ${user}`;
         
-        // Refresh statistik z DB na pozadí
-        const grUser = document.querySelector("#gr-auth-user textarea");
-        const grPass = document.querySelector("#gr-auth-pass textarea");
-        if (grUser && grPass) {
-            grUser.value = user;
-            grUser.dispatchEvent(new Event("input", { bubbles: true }));
-            grPass.value = "SESSION_REFRESH";
-            grPass.dispatchEvent(new Event("input", { bubbles: true }));
-            document.querySelector("#gr-login-btn").click();
+        if (user.toLowerCase() === "guest") {
+            document.getElementById("user-display").className = "text-sm font-medium text-amber-400";
+            document.getElementById("user-display").innerText = "👤 Free Plan (Neregistrován)";
+            document.getElementById("st-docs").innerText = guestDocsCount + " (Free)";
+            document.getElementById("st-fields").innerText = guestFieldsCount + " (Místní)";
+        } else {
+            document.getElementById("user-display").className = "text-sm font-medium text-blue-400";
+            document.getElementById("user-display").innerText = `👤 ${user}`;
+            
+            // Refresh cloudu z DB pro registrované
+            const grUser = document.querySelector("#gr-auth-user textarea");
+            const grPass = document.querySelector("#gr-auth-pass textarea");
+            if (grUser && grPass) {
+                grUser.value = user;
+                grUser.dispatchEvent(new Event("input", { bubbles: true }));
+                grPass.value = "SESSION_REFRESH";
+                grPass.dispatchEvent(new Event("input", { bubbles: true }));
+                document.querySelector("#gr-login-btn").click();
+            }
+            if(authPolling) clearInterval(authPolling);
+            authPolling = setInterval(checkAuthResponse, 150);
         }
-        if(authPolling) clearInterval(authPolling);
-        authPolling = setInterval(checkAuthResponse, 150);
     } else {
         document.getElementById("auth-screen").style.display = "flex";
         document.getElementById("app-screen").style.display = "none";
     }
 }
 
+function bypassAuthAsGuest() {
+    sessionStorage.setItem("omniparse_user", "Guest");
+    showToast("Vstupuješ jako host v bezplatném režimu.");
+    checkLoginState();
+}
+
 function handleAuthAction(action) {
     const userVal = document.getElementById("auth-username").value.trim();
     const passVal = document.getElementById("auth-password").value;
@@ -671,6 +713,7 @@ function checkAuthResponse() {
                     showToast("Přihlášení úspěšné!");
                     document.getElementById("auth-screen").style.display = "none";
                     document.getElementById("app-screen").style.display = "block";
+                    document.getElementById("user-display").className = "text-sm font-medium text-blue-400";
                     document.getElementById("user-display").innerText = `👤 ${data.username}`;
                     if (data.stats) {
                         document.getElementById("st-docs").innerText = data.stats.docs_parsed;
@@ -719,7 +762,7 @@ function executeExtraction() {
     document.getElementById("btn-loading").style.display = "flex";
     document.getElementById("status-text").innerText = "Těžím data...";
     
-    const user = sessionStorage.getItem("omniparse_user") || "";
+    const user = sessionStorage.getItem("omniparse_user") || "Guest";
     const grImg = document.querySelector("#gr-image-input textarea");
     const grUser = document.querySelector("#gr-current-user textarea");
     
@@ -756,12 +799,27 @@ function checkParseResponse() {
                 document.getElementById("st-time").innerText = `${res.parse_time_seconds}s`;
                 document.getElementById("st-conf").innerText = `${Math.round(res.security_and_compliance.confidence_score * 100)}%`;
                 document.getElementById("comp-val").innerText = res.security_and_compliance.cross_field_validation;
-                document.getElementById("comp-dup").innerText = res.security_and_compliance.duplicate_detected ? "⚠️ POTENCIÁLNÍ DUPLICITA" : "✅ OK (Unikátní)";
                 
-                if (res.user_stats) {
-                    document.getElementById("st-docs").innerText = res.user_stats.docs_parsed;
-                    document.getElementById("st-fields").innerText = res.user_stats.fields_extracted;
+                // Zobrazení duplicity
+                if (sessionStorage.getItem("omniparse_user").toLowerCase() === "guest") {
+                    document.getElementById("comp-dup").innerText = "🔒 Vyžaduje registraci";
+                } else {
+                    document.getElementById("comp-dup").innerText = res.security_and_compliance.duplicate_detected ? "⚠️ POTENCIÁLNÍ DUPLICITA" : "✅ OK (Unikátní)";
                 }
+                
+                // Počítání a update statistik na panelu
+                if (sessionStorage.getItem("omniparse_user").toLowerCase() === "guest") {
+                    guestDocsCount++;
+                    guestFieldsCount += Object.keys(res.extracted_data).length;
+                    document.getElementById("st-docs").innerText = guestDocsCount + " (Free)";
+                    document.getElementById("st-fields").innerText = guestFieldsCount + " (Místní)";
+                } else {
+                    if (res.user_stats) {
+                        document.getElementById("st-docs").innerText = res.user_stats.docs_parsed;
+                        document.getElementById("st-fields").innerText = res.user_stats.fields_extracted;
+                    }
+                }
+
                 if (res._csv_b64) {
                     csvB64Data = res._csv_b64;
                     document.getElementById("export-actions").style.display = "block";
@@ -834,7 +892,6 @@ function checkChatResponse() {
     }
 }
 
-// Inicializační kontrola session při bootu
 setTimeout(checkLoginState, 500);
 </script>
 </body>
@@ -843,10 +900,9 @@ setTimeout(checkLoginState, 500);
 
 # ─── GRADIO APPLICATION CONTAINER ───────────────────────────────────────────
 with gr.Blocks(css=CUSTOM_CSS, title="OmniParse AI Enterprise Terminal") as demo:
-    # Vstřikování custom HTML frontendu s kompletní logikou
     gr.HTML(HTML_UI)
 
-    # Skryté ovládací prvky (Bridge) — JS komunikuje přímo s těmito objekty
+    # Skryté ovládací prvky (Bridge)
     with gr.Row(elem_id="hidden-controls", visible=False):
         with gr.Column():
             gr_image_input = gr.Textbox(label="image_b64", elem_id="gr-image-input", visible=False)
@@ -859,14 +915,12 @@ with gr.Blocks(css=CUSTOM_CSS, title="OmniParse AI Enterprise Terminal") as demo
             gr_chat_btn = gr.Button("chat", elem_id="gr-chat-btn", visible=False)
             gr_chat_out = gr.Textbox(label="chat_out", elem_id="gr-chat-out", visible=False)
             
-            # Auth kanály pro přihlašování a registraci
             gr_auth_user = gr.Textbox(label="auth_user", elem_id="gr-auth-user", visible=False)
             gr_auth_pass = gr.Textbox(label="auth_pass", elem_id="gr-auth-pass", visible=False)
             gr_login_btn = gr.Button("login", elem_id="gr-login-btn", visible=False)
             gr_register_btn = gr.Button("register", elem_id="gr-register-btn", visible=False)
             gr_auth_response = gr.Textbox(label="auth_response", elem_id="gr-auth-response", visible=False)
 
-    # Propojení frontend-backend eventů
     gr_parse_btn.click(
         fn=gr_parse,
         inputs=[gr_image_input, gr_current_user],