Spaces:

prthm11
/

mcp_server_test

Sleeping

App Files Files Community

prthm11 commited on Feb 26

Commit

743d1c8

verified ·

1 Parent(s): 9c5c518

Update escalation_server.py

Browse files

Files changed (1) hide show

escalation_server.py +123 -65

escalation_server.py CHANGED Viewed

@@ -190,83 +190,141 @@ async def get_teacher_info(subject: str, class_id: str) -> dict:
 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)}"
 # --- 4. MCP PROMPTS ---
 @mcp.prompt()

 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,
+    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()