Update app.py

app.py

@@ -6,6 +6,7 @@ from datetime import datetime, timedelta
 import pytz
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
+import uuid
 
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
@@ -39,6 +40,7 @@ You can ask me about:
 - My portfolio (CodeConnect, Mobile Repair App)
 - My blog posts (e.g., Laravel tips, MERN stack guides)
 - Book an appointment (e.g., 'I want an appointment on YYYY-MM-DD at HH:MM in TIMEZONE. My email is you@example.com')
+- Cancel an appointment (e.g., '!cancel appointment YYYY-MM-DD HH:MM TIMEZONE')
 - Submit feedback (e.g., '!feedback Your service is great!')
 - Use !help for commands
 """
@@ -49,23 +51,34 @@ portfolio = {
     "mobile repair app": "A mobile repair service app with MERN Stack, featuring real-time booking and tracking. See more: https://mobilerepair.mubashirdev.com"
 }
 
-# === Blog Posts (Simulated) ===
-blog_posts = [
-    {"title": "Top 10 Laravel Tips", "url": "https://mubashirdev.com/blog/laravel-tips"},
-    {"title": "Building Scalable MERN Apps", "url": "https://mubashirdev.com/blog/mern-stack"}
-]
+# === Blog Posts (Simulated Dynamic Fetch) ===
+def fetch_blog_posts():
+    # Simulate fetching from an API or website
+    return [
+        {"title": "Top 10 Laravel Tips", "url": "https://mubashirdev.com/blog/laravel-tips"},
+        {"title": "Building Scalable MERN Apps", "url": "https://mubashirdev.com/blog/mern-stack"},
+        {"title": "Vue vs Angular: A Comparison", "url": "https://mubashirdev.com/blog/vue-vs-angular"}
+    ]
 
 # === Analytics Store ===
-analytics = {"queries": 0, "appointments": 0, "feedbacks": 0}
+analytics = {"queries": 0, "appointments": 0, "feedbacks": 0, "cancellations": 0}
 
 # === Appointment Slots (Simulated) ===
-appointment_slots = {}  # Format: {(date, time, timezone): email}
+appointment_slots = {}  # Format: {(date, time, timezone): (email, session_id)}
+
+# === Session Management ===
+sessions = {}  # Format: {session_id: {"feedback_count": int, "appointment_count": int}}
+
+# === Rate Limits ===
+FEEDBACK_LIMIT = 3  # Max feedback submissions per session
+APPOINTMENT_LIMIT = 2  # Max appointments per session
 
 # === Global Chat History ===
-chat_history = []
+chat_history = {}
 
 # === Regex Patterns ===
 appointment_pattern = r"appointment.*?on\s*(\d{4}-\d{2}-\d{2})\s*at\s*(\d{2}:\d{2}).*?in\s*([A-Za-z/]+).*?email\s*is\s*([\w\.-]+@[\w\.-]+)"
+cancel_pattern = r"!cancel appointment\s*(\d{4}-\d{2}-\d{2})\s*(\d{2}:\d{2})\s*([A-Za-z/]+)"
 feedback_pattern = r"!feedback\s+(.+)"
 command_pattern = r"!(\w+)"  # Matches commands like !portfolio, !blog, !help
 
@@ -103,7 +116,7 @@ def is_slot_available(date_str, time_str, timezone):
 # === Language Detection and Translation ===
 def detect_and_translate(text, target_lang="en"):
     try:
-        detected_lang = single_detection(text, api_key=None)  # No API key needed for GoogleTranslator
+        detected_lang = single_detection(text, api_key=None)
         if detected_lang != target_lang:
             translated = GoogleTranslator(source="auto", target=target_lang).translate(text)
             return translated, detected_lang
@@ -113,12 +126,18 @@ def detect_and_translate(text, target_lang="en"):
         return text, "en"
 
 # === Chatbot Core ===
-def chat(user_message, user_lang="en"):
+def chat(user_message, user_lang="en", session_id=None):
     global chat_history
 
     if not user_message.strip():
         return "Please provide a valid message."
 
+    # Initialize session if none provided
+    if not session_id:
+        session_id = str(uuid.uuid4())
+        sessions[session_id] = {"feedback_count": 0, "appointment_count": 0}
+        chat_history[session_id] = []
+
     # Track query
     track_analytics("queries")
 
@@ -132,27 +151,48 @@ def chat(user_message, user_lang="en"):
         if command == "portfolio":
             return "\n".join([f"{key.title()}: {value}" for key, value in portfolio.items()])
         elif command == "blog":
-            return "\n".join([f"{post['title']}: {post['url']}" for post in blog_posts])
+            return "\n".join([f"{post['title']}: {post['url']}" for post in fetch_blog_posts()])
         elif command == "help":
-            return "Available commands:\n!portfolio - View my projects\n!blog - See my blog posts\n!feedback <message> - Submit feedback\nOr ask about services or book an appointment!"
+            return "Available commands:\n!portfolio - View my projects\n!blog - See my blog posts\n!feedback <message> - Submit feedback\n!cancel appointment YYYY-MM-DD HH:MM TIMEZONE - Cancel an appointment\nOr ask about services or book an appointment!"
         elif command == "analytics":
-            return f"Analytics: {analytics['queries']} queries, {analytics['appointments']} appointments, {analytics['feedbacks']} feedbacks"
+            return f"Analytics: {analytics['queries']} queries, {analytics['appointments']} appointments, {analytics['feedbacks']} feedbacks, {analytics['cancellations']} cancellations"
 
     # Handle feedback
     feedback_match = re.match(feedback_pattern, translated_message.lower())
     if feedback_match:
+        if sessions[session_id]["feedback_count"] >= FEEDBACK_LIMIT:
+            return "❌ Feedback limit reached for this session. Please try again later."
         feedback = feedback_match.group(1)
+        sessions[session_id]["feedback_count"] += 1
         track_analytics("feedbacks")
-        send_email("admin@mubashirdev.com", "📢 New Feedback", f"Feedback: {feedback}\nFrom: Anonymous")
+        send_email("admin@mubashirdev.com", "📢 New Feedback", f"Feedback: {feedback}\nFrom: Session {session_id}")
         return "Thank you for your feedback! It has been recorded."
 
+    # Handle appointment cancellation
+    cancel_match = re.match(cancel_pattern, translated_message.lower())
+    if cancel_match:
+        date_str, time_str, timezone = cancel_match.groups()
+        try:
+            tz = pytz.timezone(timezone)
+            slot_key = (date_str, time_str, timezone)
+            if slot_key in appointment_slots:
+                email, _ = appointment_slots[slot_key]
+                del appointment_slots[slot_key]
+                track_analytics("cancellations")
+                send_email(email, "❌ Appointment Cancelled", f"Your appointment on {date_str} at {time_str} ({timezone}) has been cancelled.")
+                return f"✅ Appointment on {date_str} at {time_str} ({timezone}) has been cancelled."
+            else:
+                return "❌ No appointment found for the specified date and time."
+        except pytz.exceptions.UnknownTimeZoneError:
+            return "❌ Invalid time zone. Please use a valid time zone (e.g., Asia/Karachi)."
+
     # Add context + message to input
     full_input = f"{context}\nUser: {translated_message}\nBot:"
     new_input = tokenizer.encode(full_input + tokenizer.eos_token, return_tensors="pt").to(torch.long)
 
-    # Manage chat history (limit to 1000 tokens)
-    if chat_history:
-        bot_input = torch.cat([torch.tensor(chat_history[-1000:]), new_input], dim=-1)
+    # Manage chat history (limit to 1000 tokens per session)
+    if session_id in chat_history and chat_history[session_id]:
+        bot_input = torch.cat([torch.tensor(chat_history[session_id][-1000:]), new_input], dim=-1)
     else:
         bot_input = new_input
 
@@ -172,11 +212,15 @@ def chat(user_message, user_lang="en"):
         reply = "Sorry, I encountered an issue while generating a response. Please try again."
 
     # Update chat history
-    chat_history.append(new_input[0].tolist())
+    if session_id not in chat_history:
+        chat_history[session_id] = []
+    chat_history[session_id].append(new_input[0].tolist())
 
     # === Appointment Booking Logic ===
     match = re.search(appointment_pattern, translated_message.lower())
     if match:
+        if sessions[session_id]["appointment_count"] >= APPOINTMENT_LIMIT:
+            return "❌ Appointment limit reached for this session. Please try again later."
         date_str, time_str, timezone, email = match.groups()
         try:
             tz = pytz.timezone(timezone)
@@ -187,10 +231,11 @@ def chat(user_message, user_lang="en"):
                 reply += "\n❌ This time slot is already booked. Please choose another time."
             else:
                 # Reserve slot
-                appointment_slots[(date_str, time_str, timezone)] = email
+                appointment_slots[(date_str, time_str, timezone)] = (email, session_id)
+                sessions[session_id]["appointment_count"] += 1
                 track_analytics("appointments")
                 # Send email to admin & user
-                admin_msg = f"📅 New Appointment Request:\nDate: {date_str}\nTime: {time_str}\nTime Zone: {timezone}\nEmail: {email}"
+                admin_msg = f"📅 New Appointment Request:\nDate: {date_str}\nTime: {time_str}\nTime Zone: {timezone}\nEmail: {email}\nSession: {session_id}"
                 user_msg = f"""Hi,\n\n✅ Your appointment with Mubashir is confirmed for {date_str} at {time_str} ({timezone}).\n\nRegards,\nMubashir Hussain\nhttps://mubashirdev.com"""
 
                 if send_email("admin@mubashirdev.com", "📅 New Appointment", admin_msg) and send_email(
@@ -212,18 +257,24 @@ def chat(user_message, user_lang="en"):
         except Exception as e:
             print(f"[Translation Error] {e}")
 
-    return reply
+    return reply, session_id
 
 # === Gradio UI ===
-with gr.Blocks(theme=gr.themes.Soft()) as demo:
+with gr.Blocks(theme=gr.themes.Soft(), css="""
+    .gradio-container { max-width: 800px; margin: auto; }
+    .markdown { font-family: Arial, sans-serif; }
+    .chatbot { border: 1px solid #ddd; border-radius: 8px; padding: 10px; }
+    .gr-button { font-size: 16px; }
+""") as demo:
     gr.Markdown(
         """
         # Mubashir Dev Chatbot
         Ask about my services, portfolio, blog posts, or book an appointment.  
         Example: *I want an appointment on 2025-06-01 at 14:00 in Asia/Karachi. My email is you@example.com*  
-        Commands: *!portfolio*, *!blog*, *!feedback <message>*, *!help*
+        Commands: *!portfolio*, *!blog*, *!feedback <message>*, *!cancel appointment YYYY-MM-DD HH:MM TIMEZONE*, *!help*
         """
     )
+    session_id = gr.State(value=None)
     chatbot = gr.ChatInterface(
         fn=chat,
         additional_inputs=[
@@ -231,11 +282,9 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 choices=["en", "ur", "fr", "es"],
                 label="Select Language",
                 value="en"
-            )
-        ],
-        submit_btn="🚀 Send",
-        retry_btn="🔁 Retry",
-        clear_btn="🗑️ Clear",
+            ),
+            gr.State(value=session_id)
+        ]
     )
 
 # === Launch ===