Update app.py

app.py

@@ -1,58 +1,108 @@
-import gradio as gr
+from mcp.server.fastmcp import FastMCP
+from typing import Optional
+import requests
 import os
-import google.generativeai as genai
-from zoho_client_mcp import mcp # Import the MCP instance from our tools file
-
-# --- 1. Gemini Client Initialization ---
-# We configure the Gemini client directly in the app before launching the MCP.
-# This assumes os.environ['GOOGLE_API_KEY'] is set in config.py.
-def configure_gemini():
-    """Initializes the Gemini client using the API key from environment variables."""
-    try:
-        genai.configure(api_key=os.environ.get('GOOGLE_API_KEY'))
-        print("Gemini API configured successfully for conversational use.")
-    except Exception as e:
-        print(f"Error configuring Gemini API: {e}")
-
-# --- 2. Gradio Chat Interface Definition ---
-# This defines the web UI required for conversation and file uploads [3, 13, 16, 17].
+import gradio as gr
+
+from config import CLIENT_ID, CLIENT_SECRET, REFRESH_TOKEN, API_BASE
+
+# --- Initialize the FastMCP Server ---
+mcp = FastMCP("ZohoCRMAgent")
+
+# --- Token Refresh Utility ---
+def _get_valid_token_headers() -> dict:
+    """Internal function to ensure a valid Zoho access token is available.
+    This uses the refresh token flow to retrieve a fresh access token."""
+    token_url = "https://accounts.zoho.in/oauth/v2/token"
+    params = {
+        "refresh_token": REFRESH_TOKEN,
+        "client_id": CLIENT_ID,
+        "client_secret": CLIENT_SECRET,
+        "grant_type": "refresh_token"
+    }
+    response = requests.post(token_url, params=params)
+    if response.status_code == 200:
+        access_token = response.json().get("access_token")
+        return {"Authorization": f"Zoho-oauthtoken {access_token}"}
+    else:
+        raise Exception(f"Failed to refresh token: {response.text}")
+
+# --- MCP Tools for Zoho CRM and Zoho Books Operations ---
+
+@mcp.tool()
+def authenticate_zoho() -> str:
+    """Refreshes and confirms Zoho CRM access token availability."""
+    _ = _get_valid_token_headers()
+    return "Zoho CRM access token successfully refreshed."
+
+@mcp.tool()
+def create_record(module_name: str, record_data: dict) -> str:
+    """Creates a new record in the specified Zoho CRM module."""
+    headers = _get_valid_token_headers()
+    response = requests.post(f"{API_BASE}/{module_name}", headers=headers, json={"data": [record_data]})
+    if response.status_code in [200, 201]:
+        return f"Record created successfully in {module_name}."
+    return f"Error creating record: {response.text}"
+
+@mcp.tool()
+def get_records(module_name: str, page: int = 1, per_page: int = 200) -> list:
+    """Fetches records from a specified Zoho CRM module."""
+    headers = _get_valid_token_headers()
+    params = {"page": page, "per_page": per_page}
+    response = requests.get(f"{API_BASE}/{module_name}", headers=headers, params=params)
+    if response.status_code == 200:
+        return response.json().get("data", [])
+    return [f"Error retrieving records: {response.text}"]
+
+@mcp.tool()
+def update_record(module_name: str, record_id: str, data: dict) -> str:
+    """Updates a record in a Zoho CRM module."""
+    headers = _get_valid_token_headers()
+    response = requests.put(f"{API_BASE}/{module_name}/{record_id}", headers=headers, json={"data": [data]})
+    if response.status_code == 200:
+        return f"Record {record_id} in {module_name} updated successfully."
+    return f"Error updating record: {response.text}"
+
+@mcp.tool()
+def delete_record(module_name: str, record_id: str) -> str:
+    """Deletes a record from the specified Zoho CRM module."""
+    headers = _get_valid_token_headers()
+    response = requests.delete(f"{API_BASE}/{module_name}/{record_id}", headers=headers)
+    if response.status_code == 200:
+        return f"Record {record_id} in {module_name} deleted."
+    return f"Error deleting record: {response.text}"
+
+@mcp.tool()
+def create_invoice(data: dict) -> str:
+    """Creates an invoice in Zoho Books."""
+    headers = _get_valid_token_headers()
+    response = requests.post(f"{API_BASE}/invoices", headers=headers, json={"data": [data]})
+    if response.status_code in [200, 201]:
+        return "Invoice created successfully."
+    return f"Error creating invoice: {response.text}"
+
+@mcp.tool()
+def process_document(file_path: str, target_module: Optional[str] = "Contacts") -> dict:
+    """Extracts data from uploaded file (PDF/image) and returns structured info."""
+    # Placeholder for OCR + Gemini parsing logic
+    # raw_text = perform_ocr(file_path)
+    # structured_data = gemini_parse_json(raw_text)
+    return {
+        "status": "success",
+        "file": os.path.basename(file_path),
+        "extracted_data": f"Simulated structured data from {target_module} document."
+    }
+
+# --- Launch UI and MCP Server ---
 def chat_interface():
-    # Gradio's gr.ChatInterface is used to build a standard chat UI.
-    # The submission logic is handled automatically by FastMCP when mcp_server=True is used.
-    
-    # The file input is added directly to allow attachments (images/PDFs) [13, 18].
-    # Files uploaded here will be available to the 'process_document' tool [18].
-    file_component = gr.File(label="Upload Image or PDF Document", type="filepath")
-
-    # The gr.ChatInterface sets up the UI structure for the conversation.
-    iface = gr.ChatInterface(
-        # The function passed here is typically the conversational engine.
-        # When mcp_server=True, the MCP framework handles the routing.
-        fn=lambda msg, history: f"Thinking...", 
-        chatbot=gr.Chatbot(height=400),
-        textbox=gr.Textbox(placeholder="Ask me to create a contact, search records, or upload a document...", scale=7),
-        title="Zoho CRM Agent (Gemini + FastMCP)",
-        submit_btn="Send Command",
-        # We attach the file component to the interface so users can upload [3, 16].
-        additional_inputs=[file_component] 
+    return gr.ChatInterface(
+        lambda msg: f"Received: {msg}",
+        textbox=gr.Textbox(placeholder="Ask me to create contacts, deals, or upload documents."),
     )
-    return iface
-
-# --- 3. Server Launch ---
 
 if __name__ == "__main__":
-    # Ensure Gemini is ready [11, 19].
-    configure_gemini()
-    
-    # Get the defined Gradio UI
     demo = chat_interface()
-    
-    # Launch the Gradio UI, instructing it to also start the FastMCP server [4].
-    # Gradio automatically connects the UI to the running 'mcp' instance when mcp_server=True [3].
-    print("Starting Gradio Web UI and FastMCP Server...")
     demo.launch(
-        server_name="0.0.0.0", 
-        server_port=7860,
-        mcp_server=True,  # Crucial: Starts the FastMCP server instance 'mcp' [4].
-        mcp=mcp            
-    )
+        server_name="0.0.0.0",
+        server_port=7860
+    )