Update app.py

app.py

@@ -9,7 +9,7 @@ import re
 from datetime import datetime
 import os
 
-# Load SMTP credentials from Hugging Face Secrets
+# Load SMTP credentials securely
 SMTP_SERVER = "mail.mubashirdev.com"
 SMTP_PORT = 465
 SMTP_USERNAME = os.getenv("SMTP_USERNAME")
@@ -17,7 +17,7 @@ SMTP_PASSWORD = os.getenv("SMTP_PASSWORD")
 FROM_EMAIL = SMTP_USERNAME
 FROM_NAME = "Mubashir Hussain"
 
-# Send email
+# Email sender
 def send_email(to_email, subject, body):
     try:
         msg = MIMEMultipart()
@@ -32,54 +32,86 @@ def send_email(to_email, subject, body):
             server.send_message(msg)
         return True
     except Exception as e:
-        print(f"Email error: {e}")
+        print(f"[Email Error]: {e}")
         return False
 
 # Load model
-tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-large")
-model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-large")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
-# Chat context
+# Chatbot persona
 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:
+Ask me about:
 - My services
 - My portfolio (CodeConnect, Mobile Repair App)
 - My blog posts
 - Or book an appointment!
 """
 
-chat_history = []
-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\.-]+)"
+# Chat session history (Gradio state-compatible)
+def respond(message, chat_history=[]):
+    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\.-]+)"
 
-# Chat function
-def chat(message):
-    global chat_history
-    full_input = f"{context}\nUser: {message}\nBot:"
-    new_input = tokenizer.encode(full_input + tokenizer.eos_token, return_tensors="pt")
-    bot_input = new_input if not chat_history else torch.cat([torch.LongTensor(chat_history), new_input], dim=-1)
-    output = model.generate(bot_input, max_length=500, pad_token_id=tokenizer.eos_token_id)
-    reply = tokenizer.decode(output[0], skip_special_tokens=True).split("Bot:")[-1].strip()
-    chat_history.append(new_input[0].tolist())
+    # Construct full prompt
+    full_input = f"{context}\n" + "\n".join(
+        [f"User: {u}\nBot: {b}" for u, b in chat_history]
+    ) + f"\nUser: {message}\nBot:"
 
-    # Handle appointments
+    input_ids = tokenizer.encode(full_input + tokenizer.eos_token, return_tensors="pt")
+    output_ids = model.generate(
+        input_ids,
+        max_length=1000,
+        pad_token_id=tokenizer.eos_token_id,
+        temperature=0.7,
+        top_p=0.95,
+        top_k=50
+    )
+
+    output_text = tokenizer.decode(output_ids[0], skip_special_tokens=True)
+    reply = output_text.split("Bot:")[-1].strip()
+
+    # Handle appointment
     match = re.search(appointment_pattern, message.lower())
     if match:
         date_str, time_str, email = match.groups()
         try:
-            appointment_time = datetime.strptime(f"{date_str} {time_str}", "%Y-%m-%d %H:%M")
-            admin_msg = f"New Appointment:\nDate: {date_str}\nTime: {time_str}\nEmail: {email}"
+            datetime.strptime(f"{date_str} {time_str}", "%Y-%m-%d %H:%M")
+
+            # Send emails
+            admin_msg = f"📅 New Appointment Request:\nDate: {date_str}\nTime: {time_str}\nEmail: {email}"
             user_msg = f"Hi,\n\nYour 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)
-            reply += f"\n\n✅ Your appointment on {date_str} at {time_str} is confirmed. A confirmation email has been sent to {email}."
-        except:
-            reply += "\n❌ Error parsing date/time. Use format: 'I want an appointment on YYYY-MM-DD at HH:MM. My email is you@example.com'."
-    return reply
+
+            reply += f"\n\n✅ Appointment on {date_str} at {time_str} is confirmed. A confirmation email was sent to {email}."
+        except Exception as e:
+            print(f"[Parse Error]: {e}")
+            reply += "\n❌ Please use the format: 'I want an appointment on YYYY-MM-DD at HH:MM. My email is you@example.com'."
+
+    chat_history.append((message, reply))
+    return chat_history, chat_history
+
+# Reset function
+def reset():
+    return [], []
 
 # Gradio UI
-gr.ChatInterface(
-    fn=chat,
-    title="Mubashir Dev Chatbot",
-    description="Ask about my services or book an appointment.\nTry: I want an appointment on 2025-06-01 at 14:00. My email is you@example.com"
-).launch()
+with gr.Blocks(theme=gr.themes.Soft()) as demo:
+    gr.Markdown("## 👋 Mubashir Dev Chatbot")
+    gr.Markdown("Ask me anything about my work, or book an appointment.\nExample: `I want an appointment on 2025-06-01 at 14:00. My email is you@example.com`")
+
+    chatbot = gr.Chatbot()
+    msg = gr.Textbox(label="Your message")
+    state = gr.State([])
+
+    with gr.Row():
+        send_btn = gr.Button("Send")
+        clear_btn = gr.Button("Reset Chat")
+
+    send_btn.click(respond, inputs=[msg, state], outputs=[chatbot, state])
+    clear_btn.click(reset, inputs=[], outputs=[chatbot, state])
+
+# Run it
+demo.launch()