Update app.py

app.py

@@ -1,185 +1,31 @@
 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
 
-# ==========================================
-# 1. CREDENTIALS
-# ==========================================
+# --- YOUR CREDENTIALS ---
 CLIENT_ID = "1000.SIMKGAO5719K0TQ0QZQ31ZU57RLFNQ"
 CLIENT_SECRET = "60b329b4fe51930abee900cba6524ec7332cd67e06"
-REFRESH_TOKEN = "1000.fcbd3eb6544c48591ceb462eefe439e1.e6dacb8875fec5fa150596fa44eb9326"
-ORGANIZATION_ID = "60058860935" 
-
-# REGION: India
-AUTH_URL = "https://accounts.zoho.in/oauth/v2/token"
-API_BASE = "https://www.zohoapis.in/books/v3"
-
-# AI MODEL
-MODEL_ID = "Qwen/Qwen2.5-0.5B-Instruct"
-
-# ==========================================
-# 2. ZOHO CONNECTIVITY
-# ==========================================
-def get_zoho_headers():
-    try:
-        print("🔄 Refreshing Access 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"
-    print(f"🔎 Checking Vendor: {vendor_name}")
-    
-    # 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("🆕 Creating New Vendor...")
-    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"Zoho Contact Error: {create_resp.text}"
-
-def push_bill_to_zoho(payload):
-    """Creates a BILL (Expense) instead of Invoice (Sales)"""
-    headers, error = get_zoho_headers()
-    if error: return error
-    
-    # 1. Vendor Logic
-    vid, v_status = find_or_create_vendor(payload.get("vendor_name_raw"), headers)
-    if not vid: return f"❌ Vendor Error: {v_status}"
-    
-    # 2. Construct BILL Payload
-    # We need a Bill Number. If AI missed it, generate a random one.
-    bill_num = payload.get("invoice_number")
-    if not bill_num or str(bill_num).lower() == "null": 
-        bill_num = f"INV-{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"
+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"
     }
-    
-    # 3. Push to /bills
-    print(f"🚀 Creating Bill for {vid}...")
-    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", {})
-        return f"✅ SUCCESS! Bill Created.\nNumber: {bill.get('bill_number')}\nVendor: {v_status}\nLink: https://books.zoho.in/app/{ORGANIZATION_ID}#/bills/{bill.get('bill_id')}"
+    resp = requests.post(url, data=params)
+    if resp.status_code == 200:
+        return f"✅ NEW REFRESH TOKEN:\n{resp.json().get('refresh_token')}"
     else:
-        return f"❌ Zoho Error: {resp.text}"
-
-# ==========================================
-# 3. AI & ORCHESTRATOR
-# ==========================================
-print(">>> Loading AI...")
-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:
-        # Parse the JSON String
-        json_text = tokenizer.decode(out[0]).split("```json")[1].split("```")[0].strip()
-        data = json.loads(json_text)
-        
-        # --- THE FIX: Handle List vs Dict ---
-        if isinstance(data, list):
-            if len(data) > 0: return data[0] # Return first item if it's a list
-            else: return {"vendor_name": "Unknown", "total": 0}
-            
-        return data # Return dict as is
-        
-    except Exception as e:
-        print(f"AI Parsing Error: {e}")
-        return {"vendor_name": "Unknown", "total": 0}
-
-def run(file):
-    text, score, img = perform_ocr(file)
-    data = query_ai(text)
-    
-    # Ensure data is a dict before using .get()
-    if not isinstance(data, dict):
-        data = {"vendor_name": "Unknown", "total": 0}
-
-    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}]
-    }
-    
-    result = push_bill_to_zoho(payload)
-    return img, result
+        return f"❌ Error: {resp.text}"
 
-# ==========================================
-# 4. UI
-# ==========================================
-with gr.Blocks(title="Zoho Invoice MCP") as demo:
-    gr.Markdown("## 🧾 MCP Agent -> Zoho Books (Bills)")
-    with gr.Row():
-        f_in = gr.File(label="Upload Vendor Invoice")
-        btn = gr.Button("Process Bill", variant="primary")
-    out_img = gr.Image(label="Preview", height=300)
-    out_res = gr.Textbox(label="Status")
-    
-    btn.click(run, f_in, [out_img, out_res])
+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)
 
-if __name__ == "__main__":
-    demo.launch()
+demo.launch()