Spaces:

prthm11
/

mcp_server_test

Sleeping

App Files Files Community

prthm11 commited on 20 days ago

Commit

e0500da

verified ·

1 Parent(s): 743d1c8

Update escalation_server.py

Browse files

Files changed (1) hide show

escalation_server.py +110 -124

escalation_server.py CHANGED Viewed

@@ -187,89 +187,9 @@ async def get_teacher_info(subject: str, class_id: str) -> dict:
     }
 # --- RESEND EMAIL CONFIGURATION ---
-RESEND_API_KEY = os.getenv("RESEND_API_KEY")
-# @mcp.tool()
-# async def escalate_to_teacher(
-#     teacher_name: str,
-#     subject: str,
-#     student_query: str,
-#     ai_response: str,
-#     ctx: Context = CurrentContext()
-# ) -> str:
-#     """
-#     Sends a real-time email via Resend API (HTTP-based).
-#     Works on Hugging Face Spaces where SMTP is blocked.
-#     """
-#     # 1. Fetch recipient from your existing FACULTY_DATA
-#     teacher_info = FACULTY_DATA.get(teacher_name)
-#     if not teacher_info:
-#         # Fallback to your own email for testing if teacher not found
-#         recipient = "dummyuseai@gmail.com"
-#     else:
-#         recipient = teacher_info["email"]
-#     # 2. Resend API setup
-#     url = "https://api.resend.com/emails"
-#     headers = {
-#         "Authorization": f"Bearer {RESEND_API_KEY}",
-#         "Content-Type": "application/json"
-#     }
-#     # NOTE: If you haven't verified a domain in Resend yet,
-#     # you MUST use "onboarding@resend.dev" as the 'from' address.
-#     payload = {
-#         "from": "Tutor System <pratham.lakdawala@webashlar.com>",
-#         "to": [recipient],
-#         "subject": f"🚨 Escalation: {subject} Doubt for {teacher_name}",
-#         "html": f"""
-#         <html>
-#             <body style="font-family: Arial, sans-serif; color: #333; line-height: 1.6;">
-#                 <div style="max-width: 600px; margin: auto; border: 1px solid #eee; padding: 20px;">
-#                     <h2 style="color: #d9534f; border-bottom: 2px solid #d9534f; padding-bottom: 10px;">
-#                         New Escalation: {subject}
-#                     </h2>
-#                     <p>Hello <strong>{teacher_name}</strong>,</p>
-#                     <p>A student has reached the maximum tutoring attempts. Here is the session context:</p>
-#                     <div style="background: #f9f9f9; padding: 15px; border-radius: 5px; margin: 20px 0;">
-#                         <p><strong>Student Doubt:</strong><br>{student_query}</p>
-#                         <hr style="border: 0; border-top: 1px solid #ccc;">
-#                         <p><strong>Last AI Response:</strong><br>{ai_response}</p>
-#                     </div>
-#                     <p style="font-size: 0.9em; color: #666;">
-#                         Please review this doubt and provide guidance on the student dashboard.
-#                     </p>
-#                 </div>
-#             </body>
-#         </html>
-#         """
-#     }
-#     # 3. Send the request via HTTPS (Port 443 - never blocked)
-#     async with httpx.AsyncClient() as client:
-#         try:
-#             response = await client.post(url, headers=headers, json=payload)
-#             # Resend returns 200 or 201 on success
-#             if response.status_code in [200, 201]:
-#                 # --- RESET ATTEMPTS ---
-#                 await ctx.set_state("math_attempts", 0)
-#                 # RE-ENABLE TOOLS: Using the correct 'names=' keyword
-#                 await ctx.enable_components(names=["explain_from_resource", "generate_quick_revision"])
-#                 return f"Success: {teacher_name} ({subject} Expert) has been notified via Email API."
-#             else:
-#                 # Handle API-specific errors (e.g., unauthorized, over limit)
-#                 return f"Email API Error ({response.status_code}): {response.text}"
-#         except Exception as e:
-#             return f"Critical Network Error: Could not reach Email API. Details: {str(e)}"
 @mcp.tool()
 async def escalate_to_teacher(
     teacher_name: str,
     subject: str,
@@ -277,54 +197,120 @@ async def escalate_to_teacher(
     ai_response: str,
     ctx: Context = CurrentContext()
 ) -> str:
-    """Sends a real-time email to the correct teacher by name."""
-    # Safely get email from our central registry
     teacher_info = FACULTY_DATA.get(teacher_name)
-    if not teacher_info:
-        recipient = "dummyuseai@gmail.com" # Fallback to admin
-    else:
-        recipient = teacher_info["email"]
-    msg = EmailMessage()
-    msg["Subject"] = f"🚨 Escalation: {subject} Doubt for {teacher_name}"
-    msg["From"] = SENDER_EMAIL
-    msg["To"] = recipient
-    html_content = f"""
     <html>
-    <body style="font-family: Arial, sans-serif; color: #333;">
-    <div style="background: #f9f9f9; padding: 20px; border: 1px solid #ddd;">
-    <h2 style="color: #d9534f;">Hello {teacher_name},</h2>
-    <p>A student has a <strong>{subject}</strong> doubt that requires your attention.</p>
-    <hr>
-    <p><strong>Doubt:</strong> {student_query}</p>
-    <p><strong>AI Response:</strong> {ai_response}</p>
-    </div>
-    </body>
     </html>
     """
-    msg.add_alternative(html_content, subtype="html")
-    try:
-        await aiosmtplib.send(msg,
-        hostname=SMTP_SERVER,
-        port=SMTP_PORT,
-        username=SENDER_EMAIL,
-        password=SENDER_PASSWORD,
-        start_tls=True,
-        use_tls=False,
-        timeout=30
-        )
-        # --- THE FIX: RESET LOGIC ---
-        # ctx.set_state("math_attempts", 0) # Reset counter to 0
-        await ctx.set_state("math_attempts", 0)
-        # await ctx.enable_components(tools=["explain_from_resource", "generate_quick_revision"]) # Re-enable tools
-        return f"Success: {teacher_name} ({subject} Expert) has been notified."
-    except Exception as e:
-        return f"Error sending email: {str(e)}"
 # --- 4. MCP PROMPTS ---
 @mcp.prompt()

     }
 # --- RESEND EMAIL CONFIGURATION ---
+# RESEND_API_KEY = os.getenv("RESEND_API_KEY")
+VERCEL_EMAIL_API_URL = os.getenv("VERCEL_EMAIL_API_URL", "https://smtp-email-service-a61d.vercel.app/api/send_email")
 @mcp.tool()
 async def escalate_to_teacher(
     teacher_name: str,
     subject: str,
     ai_response: str,
     ctx: Context = CurrentContext()
 ) -> str:
+    """Sends a real-time email via the Vercel SMTP Bridge API."""
+    # 1. Get recipient from registry
     teacher_info = FACULTY_DATA.get(teacher_name)
+    recipient = teacher_info["email"] if teacher_info else "dummyuseai@gmail.com"
+    # 2. Prepare the email content
+    email_subject = f"🚨 Escalation: {subject} Doubt for {teacher_name}"
+    html_body = f"""
     <html>
+        <body style="font-family: Arial, sans-serif; color: #333;">
+            <div style="background: #f9f9f9; padding: 20px; border: 1px solid #ddd;">
+                <h2 style="color: #d9534f;">Hello {teacher_name},</h2>
+                <p>A student has a <strong>{subject}</strong> doubt that requires your attention.</p>
+                <hr>
+                <p><strong>Doubt:</strong> {student_query}</p>
+                <p><strong>AI Response:</strong> {ai_response}</p>
+            </div>
+        </body>
     </html>
     """
+    # 3. Construct the payload for your Vercel API
+    # Based on your index.py: to, subject, body, is_html
+    api_payload = {
+        "to": recipient,
+        "subject": email_subject,
+        "body": html_body,
+        "is_html": True
+    }
+    # 4. Call the Vercel API via HTTPS
+    async with httpx.AsyncClient() as client:
+        try:
+            response = await client.post(
+                VERCEL_EMAIL_API_URL,
+                json=api_payload,
+                timeout=30.0
+            )
+            if response.status_code == 200:
+                # --- SUCCESS: RESET LOGIC ---
+                await ctx.set_state("math_attempts", 0)
+                # Re-enable tutoring tools if they were disabled
+                await ctx.enable_components(names=["explain_from_resource", "generate_quick_revision"])
+                return f"Success: {teacher_name} ({subject} Expert) has been notified via Vercel Bridge."
+            else:
+                # Handle API-level errors (400, 401, 502, etc.)
+                error_data = response.json()
+                return f"Vercel API Error ({response.status_code}): {error_data.get('error', 'Unknown error')}"
+        except httpx.ConnectError:
+            return "Critical Error: Could not connect to Vercel API. Check your VERCEL_EMAIL_API_URL."
+        except Exception as e:
+            return f"Unexpected Error: {str(e)}"
+# @mcp.tool()
+# async def escalate_to_teacher(
+#     teacher_name: str,
+#     subject: str,
+#     student_query: str,
+#     ai_response: str,
+#     ctx: Context = CurrentContext()
+# ) -> str:
+#     """Sends a real-time email to the correct teacher by name."""
+#     # Safely get email from our central registry
+#     teacher_info = FACULTY_DATA.get(teacher_name)
+#     if not teacher_info:
+#         recipient = "dummyuseai@gmail.com" # Fallback to admin
+#     else:
+#         recipient = teacher_info["email"]
+#     msg = EmailMessage()
+#     msg["Subject"] = f"🚨 Escalation: {subject} Doubt for {teacher_name}"
+#     msg["From"] = SENDER_EMAIL
+#     msg["To"] = recipient
+#     html_content = f"""
+#     <html>
+#     <body style="font-family: Arial, sans-serif; color: #333;">
+#     <div style="background: #f9f9f9; padding: 20px; border: 1px solid #ddd;">
+#     <h2 style="color: #d9534f;">Hello {teacher_name},</h2>
+#     <p>A student has a <strong>{subject}</strong> doubt that requires your attention.</p>
+#     <hr>
+#     <p><strong>Doubt:</strong> {student_query}</p>
+#     <p><strong>AI Response:</strong> {ai_response}</p>
+#     </div>
+#     </body>
+#     </html>
+#     """
+#     msg.add_alternative(html_content, subtype="html")
+#     try:
+#         await aiosmtplib.send(msg,
+#         hostname=SMTP_SERVER,
+#         port=SMTP_PORT,
+#         username=SENDER_EMAIL,
+#         password=SENDER_PASSWORD,
+#         start_tls=True,
+#         use_tls=False,
+#         timeout=30
+#         )
+#         # --- THE FIX: RESET LOGIC ---
+#         # ctx.set_state("math_attempts", 0) # Reset counter to 0
+#         await ctx.set_state("math_attempts", 0)
+#         # await ctx.enable_components(tools=["explain_from_resource", "generate_quick_revision"]) # Re-enable tools
+#         return f"Success: {teacher_name} ({subject} Expert) has been notified."
+#     except Exception as e:
+#         return f"Error sending email: {str(e)}"
 # --- 4. MCP PROMPTS ---
 @mcp.prompt()