Update app.py

app.py

@@ -1,335 +1,45 @@
-import streamlit as st
+import gradio as gr
 import pandas as pd
 import requests
 import json
-import base64
-from io import BytesIO
 import openpyxl
-import os
-
-# 📌 0. Hugging Face ScriptRunContext Patch (To avoid runtime warnings)
-from streamlit.runtime.scriptrunner_utils.script_run_context import add_script_run_ctx
-import threading
-
-# 📌 1. Mukhya Page Configuration
-st.set_page_config(
-    page_title="CK Export Invoice Processor Pro", 
-    page_icon="🚢",
-    layout="wide", 
-    initial_sidebar_state="expanded"
-)
+import pdfplumber
+import re
+import spaces
 
 WEB_APP_URL = "https://script.google.com/macros/s/AKfycbwEsmWdnkVW3H7_fD99vPMrqhvmY6iJHP1ZooKuwDlj2VE4cht_FBgFyem9xDRFlbjuNw/exec"
 SPREADSHEET_ID = "182qRuH7R0jZqWVKHCg_oAG1SK5CUSkQpxVPxH2O8QUQ"
 
-def fetch_all_from_sheet():
-    """गूगल शीट से JSON डेटा, टेम्पलेट्स, API Key और Exchange Rates एक साथ फेच करता है (Session State Caching के साथ)"""
-    if "gsheet_master_cache" in st.session_state and st.session_state["gsheet_master_cache"] is not None:
-        return st.session_state["gsheet_master_cache"]
-        
-    try:
-        response = requests.get(f"{WEB_APP_URL}?action=get_data", timeout=20)
-        if response.status_code == 200:
-            res_text = response.text.strip()
-            if res_text.startswith("<"):
-                return None
-            data = response.json()
-            st.session_state["gsheet_master_cache"] = data
-            return data
-    except Exception:
-        pass
-    return None
-
-def clear_sheet_cache():
-    if "gsheet_master_cache" in st.session_state:
-        st.session_state["gsheet_master_cache"] = None
-
-def save_exchange_rates_to_sheet(ex_data):
-    try:
-        payload = {
-            "action": "save_exchange_rates",
-            "exchange_data": ex_data
-        }
-        res = requests.post(WEB_APP_URL, data=json.dumps(payload), timeout=30)
-        if res.status_code == 200:
-            clear_sheet_cache()
-            return True
-        return False
-    except Exception:
-        return False
-
-# 📌 2. Session State Initialization (Global App Lock ke liye)
-if "app_authenticated" not in st.session_state:
-    st.session_state["app_authenticated"] = False
-
-# 🔒 Agar app unlock nahi hai, toh pehle password screen dikhayein
-if not st.session_state["app_authenticated"]:
-    col1, col2, col3 = st.columns([1, 1.5, 1])
-    with col2:
-        st.markdown("<br><br>", unsafe_allow_html=True)
-        st.markdown("<h2 style='text-align: center;'>🚢 CK Export Invoice Processor Pro</h2>", unsafe_allow_html=True)
-        st.markdown("<p style='text-align: center; color: gray;'>Kripya aage badhne ke liye app ka password darj karein.</p>", unsafe_allow_html=True)
-        
-        with st.form(key="login_form"):
-            pass_input = st.text_input("Password darj karein:", type="password", key="global_lock_pwd")
-            submit_button = st.form_submit_button("Unlock App", use_container_width=True, type="primary")
-            
-            if submit_button:
-                if pass_input == "CKJ":
-                    st.session_state["app_authenticated"] = True
-                    st.rerun()
-                else:
-                    st.error("❌ Galat password! Kripya sahi password darj karein.")
-    st.stop()
-
-# ==========================================
-# 🚀 Password sahi hone ke baad dikhne wala main app code
-# ==========================================
-
-st.markdown("""
-    <style>
-        [data-testid="stSidebarCollapseButton"] { display: none !important; }
-        [data-testid="stSidebarCollapsedControl"] { display: none !important; }
-        .creator-card {
-            background: linear-gradient(135deg, #1e3c72 0%, #2a5298 100%);
-            padding: 15px;
-            border-radius: 12px;
-            color: white;
-            text-align: center;
-            box-shadow: 0 4px 10px rgba(0,0,0,0.2);
-            margin-bottom: 12px;
-            margin-top: 10px;
-        }
-        .creator-name {
-            font-size: 18px;
-            font-weight: 700;
-            margin-top: 8px;
-            margin-bottom: 2px;
-        }
-        .creator-title {
-            font-size: 12px;
-            color: #d1d8e0;
-            letter-spacing: 1px;
-            text-transform: uppercase;
-            font-weight: 600;
-        }
-        [data-testid="stMetricValue"] {
-            font-size: 18px !important;
-        }
-        [data-testid="stMetricLabel"] {
-            font-size: 12px !important;
-        }
-    </style>
-""", unsafe_allow_html=True)
-
-with st.sidebar:
-    st.markdown("<br>", unsafe_allow_html=True)
-    
-    # 1. Sabse upar: Photo
+@spaces.GPU
+def process_rate_pdf(file_obj):
+    if file_obj is None:
+        return "Kripya PDF file upload karein!"
     try:
-        st.image("ck_photo.jpg", use_container_width=True)
-    except:
-        st.info("Photo loading...")
-        
-    # 2. Uske niche: Chetan Joshi wala blue profile card
-    st.markdown("""
-        <div class="creator-card">
-            <div class="creator-name">Chetan Joshi</div>
-            <div class="creator-title">📞 +91 98253 06898</div>
-            <hr style="border-color: rgba(255,255,255,0.2); margin: 8px 0;">
-            <p style='font-size: 11px; color: #f1f2f6; margin: 0;'>
-                <b>CK Export Invoice Pro v2.0</b><br>
-                Enterprise Automation & Precision.
-            </p>
-        </div>
-    """, unsafe_allow_html=True)
-
-    # Session State & Google Sheet Fetch for Exchange Rates
-    if "exchange_rates" not in st.session_state or not isinstance(st.session_state["exchange_rates"], dict):
-        sheet_full_data = fetch_all_from_sheet()
-        sheet_ex = sheet_full_data.get("exchange_rates", None) if isinstance(sheet_full_data, dict) else None
-        
-        if sheet_ex and isinstance(sheet_ex, dict):
-            st.session_state["exchange_rates"] = sheet_ex
-        else:
-            st.session_state["exchange_rates"] = {
-                "date": "21-08-2026",
-                "rates": {"EUR": "109.8", "GBP": "128.15", "USD": "94.8"},
-                "all_rates": {}
-            }
-        
-    ex_data = st.session_state["exchange_rates"]
-    if "date" not in ex_data: ex_data["date"] = "21-08-2026"
-    if "rates" not in ex_data: ex_data["rates"] = {"EUR": "109.8", "GBP": "128.15", "USD": "94.8"}
-    if "all_rates" not in ex_data: ex_data["all_rates"] = {}
-
-    # 3. Uske niche: Teen currencies (EUR, GBP, USD) chote font mein
-    r = ex_data["rates"]
-    col_e, col_g, col_u = st.columns(3)
-    with col_e: st.metric(label="EUR", value=r.get("EUR", "109.8"))
-    with col_g: st.metric(label="GBP", value=r.get("GBP", "128.15"))
-    with col_u: st.metric(label="USD", value=r.get("USD", "94.8"))
-
-    # 4. Uske niche: Bohat Badi Size mein Effective Date (w.e.f)
-    st.markdown(f"<div style='text-align: center; margin-top: 10px; margin-bottom: 12px;'><span style='font-size: 20px; font-weight: 800; color: #00cec9; background: rgba(0, 206, 201, 0.1); padding: 4px 10px; border-radius: 6px; display: inline-block;'>📅 w.e.f: {ex_data.get('date', 'N/A')}</span></div>", unsafe_allow_html=True)
-
-    # 5. Uske niche: Customs Exchange Rates heading aur PDF uploader
-    st.markdown("##### 💱 Customs Exchange Rates")
-    
-    if "processed_ex_file_id" not in st.session_state:
-        st.session_state["processed_ex_file_id"] = None
-
-    ex_pdf = st.file_uploader("➡️ Upload Rate PDF", type=["pdf"], key="ex_pdf_sidebar")
-    if ex_pdf is not None:
-        file_signature = f"{ex_pdf.name}_{ex_pdf.size}"
-        if st.session_state["processed_ex_file_id"] != file_signature:
-            import pdfplumber
-            import re
-            try:
-                with pdfplumber.open(ex_pdf) as pdf:
-                    text = ""
-                    for page in pdf.pages:
-                        t = page.extract_text()
-                        if t: text += t + "\n"
-                
-                date_match = re.search(r"w\.e\.f[\s\.:]*([\d]{2}[\-\/][\d]{2}[\-\/][\d]{4})", text, re.IGNORECASE)
-                if not date_match:
-                    date_match = re.search(r"w\.e\.f[\s\.:]*([0-9A-Za-z\-]+)", text, re.IGNORECASE)
-                    
-                if date_match:
-                    ex_data["date"] = date_match.group(1).strip()
-                    
-                lines = text.split("\n")
-                all_parsed = {}
-                for line in lines:
-                    parts = line.split()
-                    if len(parts) >= 5 and parts[0].isupper() and len(parts[0]) == 3:
-                        curr_code = parts[0]
-                        export_val = parts[-1]
-                        try:
-                            float(export_val)
-                            all_parsed[curr_code] = export_val
-                        except:
-                            pass
-                
-                if all_parsed:
-                    ex_data["all_rates"] = all_parsed
-                    for c in ["EUR", "GBP", "USD"]:
-                        if c in all_parsed:
-                            ex_data["rates"][c] = all_parsed[c]
-                    
-                    save_exchange_rates_to_sheet(ex_data)
-                    st.session_state["processed_ex_file_id"] = file_signature
-                    
-                    st.success(f"🎉 Rates updated & saved to Sheet (w.e.f {ex_data['date']})!")
-                    st.rerun()
-                else:
-                    st.warning("⚠️ PDF se data nahi padha ja saka!")
-            except Exception as e:
-                st.error(f"Error: {str(e)}")
-
-    # Expander to view all other currency rates
-    if ex_data["all_rates"]:
-        with st.expander("📉 View All Currencies"):
-            for c_code, c_val in ex_data["all_rates"].items():
-                st.markdown(f"**{c_code}**: `{c_val}`")
-
-    st.markdown("---")
-    
-    if st.button("🔒 Lock App", use_container_width=True):
-        st.session_state["app_authenticated"] = False
-        st.rerun()
-        
-    st.markdown("---")
-
-@st.cache_data(show_spinner=False)
-def load_data_from_gsheet():
-    """Sidhe 'Shipper_JSON_Database' sheet se sara data load karta hai"""
-    shipper_db = {}
-    master_rules_template = {}
-    
-    try:
-        json_db_url = f"https://docs.google.com/spreadsheets/d/{SPREADSHEET_ID}/gviz/tq?tqx=out:csv&sheet=Shipper_JSON_Database"
-        df_json = pd.read_csv(json_db_url)
-        if not df_json.empty:
-            df_json.columns = df_json.columns.str.strip()
-            for _, row in df_json.iterrows():
-                if "ShipperName" in df_json.columns and pd.notna(row["ShipperName"]):
-                    s_name = str(row["ShipperName"]).strip()
-                    if s_name and s_name.lower() != "nan":
-                        shipper_json_str = row["ShipperJSON"] if "ShipperJSON" in df_json.columns and pd.notna(row["ShipperJSON"]) else "{}"
-                        try:
-                            shipper_db[s_name] = json.loads(shipper_json_str)
-                        except:
-                            shipper_db[s_name] = {
-                                "allowed_uploads": ["Full Job Excel Format File"],
-                                "uploaded_files": {},
-                                "mapping_rules": {},
-                                "item_table_rules": {},
-                                "item_table_rule_name": "parser_welspun",
-                                "igst_config": {"lut_keywords": "", "paid_keywords": ""}
-                            }
+        with pdfplumber.open(file_obj.name) as pdf:
+            text = ""
+            for page in pdf.pages:
+                t = page.extract_text()
+                if t: text += t + "\n"
+        
+        date_match = re.search(r"w\.e\.f[\s\.:]*([\d]{2}[\-\/][\d]{2}[\-\/][\d]{4})", text, re.IGNORECASE)
+        rate_date = date_match.group(1).strip() if date_match else "N/A"
+        return f"🎉 PDF Parsed Successfully! w.e.f: {rate_date}"
     except Exception as e:
-        st.error(f"Error loading Shipper_JSON_Database: {e}")
-        
-    return shipper_db, master_rules_template
-
-# 🔄 Data load engine
-if "shipper_database" not in st.session_state or "master_rules_template" not in st.session_state:
-    db, m_template = load_data_from_gsheet()
-    st.session_state["shipper_database"] = db
-    st.session_state["master_rules_template"] = m_template
-    st.session_state["master_types"] = ["Full Job Excel Format File"]
-
-if "admin_authenticated" not in st.session_state: st.session_state["admin_authenticated"] = False
-if "processed_file_ready" not in st.session_state: st.session_state["processed_file_ready"] = None
+        return f"Error: {str(e)}"
 
-# ==========================================
-# 🖥️ MAIN PAGE ROUTING DISPLAY
-# ==========================================
-if st.session_state["admin_authenticated"]:
-    top_col1, top_col2 = st.columns([8, 2])
-    with top_col1:
-        st.title("🚢 CK Export Processor - Admin Mode")
-    with top_col2:
-        if st.button("🚪 Log Out Admin", type="primary", use_container_width=True):
-            st.session_state["admin_authenticated"] = False
-            st.rerun()
-            
-    st.write("---")
-    
-    sub_menu = st.radio(
-        "📋 Admin settings chunein:", 
-        ["i. 🏢 Add Shipper Name & Setup", "iii. 🌍 Global Masters & Common Dictionaries"],
-        horizontal=True
-    )
-    st.write("---")
-    
-    from shipper_data import render_shipper_data
-    from global_masters import render_global_masters
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("# 🚢 CK Export Invoice Processor Pro")
+    gr.Markdown("ZeroGPU Enabled & Free Integration")
     
-    if sub_menu == "i. 🏢 Add Shipper Name & Setup": 
-        render_shipper_data()
-    elif sub_menu == "iii. 🌍 Global Masters & Common Dictionaries": 
-        render_global_masters()
-
-else:
-    from processor import render_processor
-    
-    col_l, col_c, col_r = st.columns([1, 5, 1])
-    with col_c:
-        st.title("🚢 CK Export Invoice Processor")
-        st.write("---")
-        render_processor()
+    with gr.Tab("Invoice Processor"):
+        gr.Markdown("### Upload your files here to process export invoices.")
+        file_input = gr.File(label="Upload Job Excel / PDF")
+        output_box = gr.Textbox(label="Processing Status")
         
-        st.write("---")
-        st.write("---")
-        
-        with st.expander("🛠️ Admin Settings Access"):
-            pwd = st.text_input("Admin password darj karein:", type="password", key="main_admin_pwd")
-            if st.button("Login Karein"):
-                if pwd == "TEST":
-                    st.session_state["admin_authenticated"] = True
-                    st.rerun()
-                else:
-                    st.error("Galat password!")
+    with gr.Tab("Customs Exchange Rates"):
+        pdf_rate_input = gr.File(label="Upload Rate PDF", file_types=[".pdf"])
+        rate_output = gr.Textbox(label="Rate Status")
+        pdf_rate_input.change(fn=process_rate_pdf, inputs=pdf_rate_input, outputs=rate_output)
+
+if __name__ == "__main__":
+    demo.launch()