Update app.py

app.py

@@ -2,41 +2,72 @@ import os
 import re
 import ssl
 import smtplib
-from datetime import datetime
+from datetime import datetime, timedelta
+import pytz
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
 
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+from googletrans import Translator, LANGUAGES
 import gradio as gr
 
 # === Email Configuration ===
 SMTP_SERVER = "mail.mubashirdev.com"
 SMTP_PORT = 465
-SMTP_USERNAME = os.getenv("SMTP_USERNAME")  # Set in HF Secrets
+SMTP_USERNAME = os.getenv("SMTP_USERNAME")  # Set in environment variables
 SMTP_PASSWORD = os.getenv("SMTP_PASSWORD")
 FROM_EMAIL = SMTP_USERNAME
 FROM_NAME = "Mubashir Hussain"
 
 # === AI Model Initialization ===
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+try:
+    tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
+    model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+except Exception as e:
+    print(f"[Model Loading Error] {e}")
+    raise
+
+# === Translation Model ===
+translator = Translator()
 
 # === Chat Context ===
 context = """
 I am Mubashir Hussain, a Full Stack Developer from Lahore, Pakistan. I specialize in Laravel, Livewire, MERN Stack, Vue, and Angular. I help startups build secure and scalable web apps.
 You can ask me about:
-- My services
+- My services (e.g., web development, API integration)
 - My portfolio (CodeConnect, Mobile Repair App)
-- My blog posts
-- Or book an appointment!
+- 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')
+- Submit feedback (e.g., '!feedback Your service is great!')
+- Use !help for commands
 """
 
+# === Portfolio Data ===
+portfolio = {
+    "codeconnect": "CodeConnect is a social platform for developers to collaborate on projects. Built with Laravel, Livewire, and Vue. Check it out: https://codeconnect.mubashirdev.com",
+    "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"}
+]
+
+# === Analytics Store ===
+analytics = {"queries": 0, "appointments": 0, "feedbacks": 0}
+
+# === Appointment Slots (Simulated) ===
+appointment_slots = {}  # Format: {(date, time, timezone): email}
+
 # === Global Chat History ===
 chat_history = []
 
-# === Regex for Appointment Booking ===
-appointment_pattern = r"appointment.*?on\s*(\d{4}-\d{2}-\d{2})\s*at\s*(\d{2}:\d{2}).*?email\s*is\s*([\w\.-]+@[\w\.-]+)"
+# === 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\.-]+)"
+feedback_pattern = r"!feedback\s+(.+)"
+command_pattern = r"!(\w+)"  # Matches commands like !portfolio, !blog, !help
 
 # === Email Utility ===
 def send_email(to_email, subject, body):
@@ -56,63 +87,160 @@ def send_email(to_email, subject, body):
         print(f"[Email Error] {e}")
         return False
 
+# === Analytics Tracking ===
+def track_analytics(event):
+    analytics[event] = analytics.get(event, 0) + 1
+
+# === Appointment Slot Check ===
+def is_slot_available(date_str, time_str, timezone):
+    try:
+        tz = pytz.timezone(timezone)
+        slot_key = (date_str, time_str, timezone)
+        return slot_key not in appointment_slots
+    except pytz.exceptions.UnknownTimeZoneError:
+        return False
+
+# === Language Detection and Translation ===
+def detect_and_translate(text, target_lang="en"):
+    try:
+        detected_lang = translator.detect(text).lang
+        if detected_lang != target_lang and target_lang in LANGUAGES:
+            translated = translator.translate(text, dest=target_lang).text
+            return translated, detected_lang
+        return text, detected_lang
+    except Exception as e:
+        print(f"[Translation Error] {e}")
+        return text, "en"
+
 # === Chatbot Core ===
-def chat(user_message):
+def chat(user_message, user_lang="en"):
     global chat_history
 
+    if not user_message.strip():
+        return "Please provide a valid message."
+
+    # Track query
+    track_analytics("queries")
+
+    # Translate input to English if needed
+    translated_message, detected_lang = detect_and_translate(user_message, "en")
+
+    # Handle commands
+    command_match = re.match(command_pattern, translated_message.lower())
+    if command_match:
+        command = command_match.group(1)
+        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])
+        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!"
+        elif command == "analytics":
+            return f"Analytics: {analytics['queries']} queries, {analytics['appointments']} appointments, {analytics['feedbacks']} feedbacks"
+
+    # Handle feedback
+    feedback_match = re.match(feedback_pattern, translated_message.lower())
+    if feedback_match:
+        feedback = feedback_match.group(1)
+        track_analytics("feedbacks")
+        send_email("admin@mubashirdev.com", "📢 New Feedback", f"Feedback: {feedback}\nFrom: Anonymous")
+        return "Thank you for your feedback! It has been recorded."
+
     # Add context + message to input
-    full_input = f"{context}\nUser: {user_message}\nBot:"
+    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
-    bot_input = new_input if not chat_history else torch.cat([torch.tensor(chat_history), new_input], dim=-1)
+    # Manage chat history (limit to 1000 tokens)
+    if chat_history:
+        bot_input = torch.cat([torch.tensor(chat_history[-1000:]), new_input], dim=-1)
+    else:
+        bot_input = new_input
 
     # Generate response
-    output = model.generate(
-        bot_input,
-        max_length=500,
-        pad_token_id=tokenizer.eos_token_id,
-        temperature=0.7,
-        top_k=50,
-        top_p=0.95,
-    )
+    try:
+        output = model.generate(
+            bot_input,
+            max_length=500,
+            pad_token_id=tokenizer.eos_token_id,
+            temperature=0.7,
+            top_k=50,
+            top_p=0.95,
+        )
+        reply = tokenizer.decode(output[0], skip_special_tokens=True).split("Bot:")[-1].strip()
+    except Exception as e:
+        print(f"[Model Generation Error] {e}")
+        reply = "Sorry, I encountered an issue while generating a response. Please try again."
 
-    reply = tokenizer.decode(output[0], skip_special_tokens=True).split("Bot:")[-1].strip()
+    # Update chat history
     chat_history.append(new_input[0].tolist())
 
     # === Appointment Booking Logic ===
-    match = re.search(appointment_pattern, user_message.lower())
+    match = re.search(appointment_pattern, translated_message.lower())
     if match:
-        date_str, time_str, email = match.groups()
+        date_str, time_str, timezone, email = match.groups()
         try:
-            datetime.strptime(f"{date_str} {time_str}", "%Y-%m-%d %H:%M")  # Validate date/time
-            # Send email to admin & user
-            admin_msg = f"📅 New Appointment Request:\nDate: {date_str}\nTime: {time_str}\nEmail: {email}"
-            user_msg = f"""Hi,\n\n✅ Your appointment with Mubashir is confirmed for {date_str} at {time_str}.\n\nRegards,\nMubashir Hussain\nhttps://mubashirdev.com"""
-
-            send_email("admin@mubashirdev.com", "📅 New Appointment", admin_msg)
-            send_email(email, "✅ Appointment Confirmed", user_msg)
+            tz = pytz.timezone(timezone)
+            appointment_time = datetime.strptime(f"{date_str} {time_str}", "%Y-%m-%d %H:%M").replace(tzinfo=tz)
+            if appointment_time < datetime.now(tz):
+                reply += "\n❌ Appointments must be scheduled in the future. Please choose a valid date and time."
+            elif not is_slot_available(date_str, time_str, timezone):
+                reply += "\n❌ This time slot is already booked. Please choose another time."
+            else:
+                # Reserve slot
+                appointment_slots[(date_str, time_str, timezone)] = email
+                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}"
+                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(
+                    email, "✅ Appointment Confirmed", user_msg
+                ):
+                    reply += f"\n\n✅ Your appointment on {date_str} at {time_str} ({timezone}) is confirmed. A confirmation email has been sent to {email}."
+                else:
+                    reply += "\n❌ Failed to send confirmation emails. Please try again later."
+        except pytz.exceptions.UnknownTimeZoneError:
+            reply += "\n❌ Invalid time zone. Please use a valid time zone (e.g., Asia/Karachi)."
+        except ValueError as e:
+            print(f"[Date Parse Error] {e}")
+            reply += "\n❌ Invalid date/time format. Please use: 'I want an appointment on YYYY-MM-DD at HH:MM in TIMEZONE. My email is you@example.com'"
 
-            reply += f"\n\n✅ Your appointment on {date_str} at {time_str} is confirmed. A confirmation email has been sent to {email}."
+    # Translate response back to user's language
+    if detected_lang != "en" and detected_lang in LANGUAGES:
+        try:
+            reply = translator.translate(reply, dest=detected_lang).text
         except Exception as e:
-            print(f"[Date Parse Error] {e}")
-            reply += "\n❌ Invalid date/time format. Please use: 'I want an appointment on YYYY-MM-DD at HH:MM. My email is you@example.com'"
+            print(f"[Translation Error] {e}")
 
     return reply
 
 # === Gradio UI ===
-demo = gr.ChatInterface(
-    fn=chat,
-    title="Mubashir Dev Chatbot",
-    description=(
-        "Ask about my services or book an appointment.\n"
-        "Example: I want an appointment on 2025-06-01 at 14:00. My email is you@example.com"
-    ),
-    theme="soft",
-    retry_btn="🔁 Retry",
-    submit_btn="🚀 Send",
-)
+with gr.Blocks(theme=gr.themes.Soft()) 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*
+        """
+    )
+    chatbot = gr.ChatInterface(
+        fn=chat,
+        additional_inputs=[
+            gr.Dropdown(
+                choices=["en", "ur", "fr", "es"],
+                label="Select Language",
+                value="en"
+            )
+        ],
+        submit_btn="🚀 Send",
+        retry_btn="🔁 Retry",
+        clear_btn="🗑️ Clear",
+    )
 
 # === Launch ===
 if __name__ == "__main__":
-    demo.launch()
+    try:
+        demo.launch(server_name="0.0.0.0", server_port=7860)
+    except Exception as e:
+        print(f"[Gradio Launch Error] {e}")