Update app.py

app.py

@@ -1,31 +1,168 @@
 import gradio as gr
+import torch
+from transformers import AutoModelForCausalLM, AutoTokenizer, TextIteratorStreamer
+from threading import Thread
+import os
+import pytesseract
+from pdf2image import convert_from_path
+from PIL import Image
+import trafilatura
+import datetime
+import json
 import requests
+import re
+import random
 
-# --- YOUR CREDENTIALS ---
+# ==========================================
+# 1. CREDENTIALS & CONFIG
+# ==========================================
 CLIENT_ID = "1000.SIMKGAO5719K0TQ0QZQ31ZU57RLFNQ"
 CLIENT_SECRET = "60b329b4fe51930abee900cba6524ec7332cd67e06"
-REDIRECT_URI = "http://www.google.com" 
-
-def get_token(code):
-    url = "https://accounts.zoho.in/oauth/v2/token"
-    params = {
-        "code": code,
-        "client_id": CLIENT_ID,
-        "client_secret": CLIENT_SECRET,
-        "redirect_uri": REDIRECT_URI,
-        "grant_type": "authorization_code"
+
+# *** PASTE YOUR NEW "FULL ACCESS" TOKEN HERE ***
+REFRESH_TOKEN = "1000.78971de3425fc7c86da12ecaedac5b58.4f3bb2afcf196ee05569ac5482570173" 
+
+# *** YOUR ORG ID ***
+ORGANIZATION_ID = "60058860935"
+
+# REGION & API
+AUTH_URL = "https://accounts.zoho.in/oauth/v2/token"
+API_BASE = "https://www.zohoapis.in/books/v3"
+MODEL_ID = "Qwen/Qwen2.5-0.5B-Instruct"
+
+# ==========================================
+# 2. ZOHO CONNECTIVITY
+# ==========================================
+def get_zoho_headers():
+    try:
+        print("🔄 Refreshing Token...")
+        params = {
+            "refresh_token": REFRESH_TOKEN,
+            "client_id": CLIENT_ID,
+            "client_secret": CLIENT_SECRET,
+            "grant_type": "refresh_token",
+            "redirect_uri": "http://www.google.com"
+        }
+        resp = requests.post(AUTH_URL, data=params)
+        if resp.status_code != 200: return None, f"Auth Failed: {resp.text}"
+        return {"Authorization": f"Zoho-oauthtoken {resp.json().get('access_token')}"}, None
+    except Exception as e:
+        return None, str(e)
+
+def find_or_create_vendor(vendor_name, headers):
+    if not vendor_name or len(str(vendor_name)) < 2: vendor_name = "MCP Vendor"
+    
+    # 1. Search
+    search_url = f"{API_BASE}/contacts"
+    try:
+        params = {"organization_id": ORGANIZATION_ID, "contact_name_contains": vendor_name, "contact_type": "vendor"}
+        resp = requests.get(search_url, headers=headers, params=params)
+        contacts = resp.json().get("contacts", [])
+        if contacts: return contacts[0]["contact_id"], "Found Existing"
+    except: pass
+
+    # 2. Create
+    print(f"🆕 Creating Vendor: {vendor_name}")
+    create_payload = {"contact_name": vendor_name, "contact_type": "vendor"}
+    create_resp = requests.post(search_url, headers=headers, params={"organization_id": ORGANIZATION_ID}, json=create_payload)
+    
+    if create_resp.status_code == 201:
+        return create_resp.json().get("contact", {}).get("contact_id"), "Created New"
+    else:
+        return None, f"Contact Error: {create_resp.text}"
+
+def push_bill_to_zoho(payload):
+    headers, error = get_zoho_headers()
+    if error: return error
+    
+    # 1. Vendor
+    vid, v_status = find_or_create_vendor(payload.get("vendor_name_raw"), headers)
+    if not vid: return f"❌ Vendor Logic Failed: {v_status}"
+    
+    # 2. Bill Payload
+    bill_num = payload.get("invoice_number")
+    if not bill_num or str(bill_num).lower() == "null": 
+        bill_num = f"MCP-{random.randint(1000,9999)}"
+
+    final_json = {
+        "vendor_id": vid,
+        "bill_number": str(bill_num),
+        "date": payload.get("date", datetime.date.today().strftime("%Y-%m-%d")),
+        "line_items": payload.get("line_items", []),
+        "status": "draft"
     }
-    resp = requests.post(url, data=params)
-    if resp.status_code == 200:
-        return f"✅ NEW REFRESH TOKEN:\n{resp.json().get('refresh_token')}"
+    
+    # 3. Push
+    url = f"{API_BASE}/bills?organization_id={ORGANIZATION_ID}"
+    resp = requests.post(url, headers=headers, json=final_json)
+    
+    if resp.status_code == 201:
+        bill = resp.json().get("bill", {})
+        link = f"https://books.zoho.in/app/{ORGANIZATION_ID}#/bills/{bill.get('bill_id')}"
+        return f"✅ SUCCESS! Bill Created.\nVendor: {v_status}\nLink: {link}"
     else:
-        return f"❌ Error: {resp.text}"
+        return f"❌ Zoho API Error: {resp.text}"
+
+# ==========================================
+# 3. AI ENGINE
+# ==========================================
+try:
+    tokenizer = AutoTokenizer.from_pretrained(MODEL_ID)
+    model = AutoModelForCausalLM.from_pretrained(MODEL_ID, device_map="cpu", torch_dtype=torch.float32, low_cpu_mem_usage=True)
+except: model = None
+
+def perform_ocr(file_obj):
+    if file_obj is None: return "", 0, None
+    try:
+        filename = os.path.basename(file_obj)
+        if filename.lower().endswith(".pdf"):
+            image = convert_from_path(file_obj, first_page=1, last_page=1)[0]
+        else:
+            image = Image.open(file_obj).convert("RGB")
+        return pytesseract.image_to_string(image), 0.9, image
+    except: return "", 0, None
+
+def query_ai(text):
+    if not model: return {"vendor_name": "Unknown", "total": 0}
+    
+    prompt = f"<|im_start|>user\nExtract JSON: vendor_name, invoice_number, invoice_date (YYYY-MM-DD), total, item_desc\nText:\n{text[:1000]}<|im_end|>\n<|im_start|>assistant\n```json"
+    inputs = tokenizer(prompt, return_tensors="pt")
+    out = model.generate(**inputs, max_new_tokens=200)
+    
+    try:
+        json_str = tokenizer.decode(out[0]).split("```json")[1].split("```")[0].strip()
+        data = json.loads(json_str)
+        return data[0] if isinstance(data, list) else data
+    except:
+        return {"vendor_name": "Unknown", "total": 0}
+
+def run(file):
+    text, score, img = perform_ocr(file)
+    data = query_ai(text)
+    
+    try: rate = float(str(data.get("total","0")).replace(",","").replace("INR","").strip())
+    except: rate = 0
+        
+    payload = {
+        "vendor_name_raw": data.get("vendor_name"),
+        "invoice_number": data.get("invoice_number"),
+        "date": data.get("invoice_date"),
+        "line_items": [{"name": data.get("item_desc", "Service"), "rate": rate}]
+    }
+    
+    return img, push_bill_to_zoho(payload)
 
-with gr.Blocks() as demo:
-    gr.Markdown("## 🔑 Zoho Token Fixer (Full Access)")
-    code_in = gr.Textbox(label="Paste New 'ZohoBooks.fullaccess.ALL' Code")
-    btn = gr.Button("Get Master Token")
-    out = gr.Textbox(label="Result")
-    btn.click(get_token, code_in, out)
+# ==========================================
+# 4. UI
+# ==========================================
+with gr.Blocks(title="Zoho Bill Agent") as demo:
+    gr.Markdown("## 🧾 MCP Agent -> Zoho Bills")
+    with gr.Row():
+        f_in = gr.File(label="Vendor Invoice")
+        btn = gr.Button("Process Bill", variant="primary")
+    out_img = gr.Image(label="View", height=300)
+    out_res = gr.Textbox(label="Status")
+    btn.click(run, f_in, [out_img, out_res])
 
-demo.launch()
+if __name__ == "__main__":
+    demo.launch()