Upload app.py with huggingface_hub

app.py

@@ -3,20 +3,49 @@ import gradio as gr
 from fastmcp import FastMCP
 from openai import OpenAI
 from memory_sync import save_to_databank, load_from_databank, get_embeddings
+import stripe
 from ebook_pipeline import create_ebook_files
 
 # Load Environment Variables
 OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY")
 HF_TOKEN = os.environ.get("HF_TOKEN")
+STRIPE_API_KEY = os.environ.get("STRIPE_API_KEY")
 
-# Initialize OpenAI Client (using GPT-4o-mini as requested)
+# Initialize Clients
 client = OpenAI(api_key=OPENAI_API_KEY)
+if STRIPE_API_KEY:
+    stripe.api_key = STRIPE_API_KEY
 
-# Initialize MCP Server
-mcp = FastMCP("Aussie Agent Hub")
+# ... (MCP Server Init)
 
 # --- MCP TOOLS ---
 
+@mcp.tool()
+def create_stripe_checkout_session(price_id: str, success_url: str, cancel_url: str) -> str:
+    """Create a Stripe Checkout Session for a given Price ID."""
+    try:
+        session = stripe.checkout.Session.create(
+            payment_method_types=['card'],
+            line_items=[{'price': price_id, 'quantity': 1}],
+            mode='payment',
+            success_url=success_url,
+            cancel_url=cancel_url,
+        )
+        return f"Checkout Session created: {session.url}"
+    except Exception as e:
+        return f"Error creating session: {str(e)}"
+
+# ... (Existing tools)
+
+# --- AGENT LOGIC ---
+
+def aussie_router(user_input, history):
+    # ...
+    # Updated router instructions logic (should ideally be loaded from databank)
+    # Adding Stripe to the dispatch logic
+    system_instr = load_from_databank("router_instructions.md")
+    # ...
+
 @mcp.tool()
 def generate_ebook(title: str, author: str, chapters: list) -> str:
     """Generate EPUB and PDF files from a list of chapters (title and content)."""