added the public url code and prompt for bot behaviour

.gradio/certificate.pem

@@ -0,0 +1,31 @@
+-----BEGIN CERTIFICATE-----
+MIIFazCCA1OgAwIBAgIRAIIQz7DSQONZRGPgu2OCiwAwDQYJKoZIhvcNAQELBQAw
+TzELMAkGA1UEBhMCVVMxKTAnBgNVBAoTIEludGVybmV0IFNlY3VyaXR5IFJlc2Vh
+cmNoIEdyb3VwMRUwEwYDVQQDEwxJU1JHIFJvb3QgWDEwHhcNMTUwNjA0MTEwNDM4
+WhcNMzUwNjA0MTEwNDM4WjBPMQswCQYDVQQGEwJVUzEpMCcGA1UEChMgSW50ZXJu
+ZXQgU2VjdXJpdHkgUmVzZWFyY2ggR3JvdXAxFTATBgNVBAMTDElTUkcgUm9vdCBY
+MTCCAiIwDQYJKoZIhvcNAQEBBQADggIPADCCAgoCggIBAK3oJHP0FDfzm54rVygc
+h77ct984kIxuPOZXoHj3dcKi/vVqbvYATyjb3miGbESTtrFj/RQSa78f0uoxmyF+
+0TM8ukj13Xnfs7j/EvEhmkvBioZxaUpmZmyPfjxwv60pIgbz5MDmgK7iS4+3mX6U
+A5/TR5d8mUgjU+g4rk8Kb4Mu0UlXjIB0ttov0DiNewNwIRt18jA8+o+u3dpjq+sW
+T8KOEUt+zwvo/7V3LvSye0rgTBIlDHCNAymg4VMk7BPZ7hm/ELNKjD+Jo2FR3qyH
+B5T0Y3HsLuJvW5iB4YlcNHlsdu87kGJ55tukmi8mxdAQ4Q7e2RCOFvu396j3x+UC
+B5iPNgiV5+I3lg02dZ77DnKxHZu8A/lJBdiB3QW0KtZB6awBdpUKD9jf1b0SHzUv
+KBds0pjBqAlkd25HN7rOrFleaJ1/ctaJxQZBKT5ZPt0m9STJEadao0xAH0ahmbWn
+OlFuhjuefXKnEgV4We0+UXgVCwOPjdAvBbI+e0ocS3MFEvzG6uBQE3xDk3SzynTn
+jh8BCNAw1FtxNrQHusEwMFxIt4I7mKZ9YIqioymCzLq9gwQbooMDQaHWBfEbwrbw
+qHyGO0aoSCqI3Haadr8faqU9GY/rOPNk3sgrDQoo//fb4hVC1CLQJ13hef4Y53CI
+rU7m2Ys6xt0nUW7/vGT1M0NPAgMBAAGjQjBAMA4GA1UdDwEB/wQEAwIBBjAPBgNV
+HRMBAf8EBTADAQH/MB0GA1UdDgQWBBR5tFnme7bl5AFzgAiIyBpY9umbbjANBgkq
+hkiG9w0BAQsFAAOCAgEAVR9YqbyyqFDQDLHYGmkgJykIrGF1XIpu+ILlaS/V9lZL
+ubhzEFnTIZd+50xx+7LSYK05qAvqFyFWhfFQDlnrzuBZ6brJFe+GnY+EgPbk6ZGQ
+3BebYhtF8GaV0nxvwuo77x/Py9auJ/GpsMiu/X1+mvoiBOv/2X/qkSsisRcOj/KK
+NFtY2PwByVS5uCbMiogziUwthDyC3+6WVwW6LLv3xLfHTjuCvjHIInNzktHCgKQ5
+ORAzI4JMPJ+GslWYHb4phowim57iaztXOoJwTdwJx4nLCgdNbOhdjsnvzqvHu7Ur
+TkXWStAmzOVyyghqpZXjFaH3pO3JLF+l+/+sKAIuvtd7u+Nxe5AW0wdeRlN8NwdC
+jNPElpzVmbUq4JUagEiuTDkHzsxHpFKVK7q4+63SM1N95R1NbdWhscdCb+ZAJzVc
+oyi3B43njTOQ5yOf+1CceWxG1bQVs5ZufpsMljq4Ui0/1lvh+wjChP4kqKOJ2qxq
+4RgqsahDYVvTH9w7jXbyLeiNdd8XM2w9U/t7y0Ff/9yi0GE44Za4rF2LN9d11TPA
+mRGunUHBcnWEvgJBQl9nJEiU0Zsnvgc/ubhPgXRR4Xq37Z0j4r7g1SgEEzwxA57d
+emyPxgcYxn/eR44/KJ4EBs+lVDR3veyJm+kXQ99b21/+jh5Xos1AnX5iItreGCc=
+-----END CERTIFICATE-----

README.md

@@ -1,8 +1,8 @@
 ---
-title: DialoGPT Chatbot
-emoji: 🤖
-colorFrom: blue
-colorTo: purple
+title: Aura - Your Supportive Friend
+emoji: 🌿
+colorFrom: green
+colorTo: blue
 sdk: gradio
 sdk_version: 3.50.2
 app_file: app.py
@@ -10,39 +10,49 @@ pinned: false
 license: mit
 ---
 
-# DialoGPT Chatbot
+# 🌿 Aura - Your Supportive Friend
 
-A conversational AI chatbot powered by Microsoft's DialoGPT-medium model, hosted on Hugging Face Spaces.
+Meet Aura, a warm, empathetic AI companion powered by Microsoft's DialoGPT-medium model. Aura is designed to be a supportive friend who listens without judgment and provides comfort during difficult times.
 
-## About
+## About Aura
 
-This chatbot uses the `microsoft/DialoGPT-medium` model, a pre-trained conversational AI model that can engage in natural dialogue. The interface is built with Gradio for easy interaction.
+Aura is not here to solve your problems or give advice unless you ask. Instead, Aura focuses on:
+
+- **Listening with empathy** and understanding
+- **Validating your feelings** and experiences  
+- **Providing a safe, non-judgmental space** to express yourself
+- **Offering gentle support** and reassurance
 
 ## Features
 
-- Natural conversation flow
-- Context-aware responses based on chat history
-- Clean and user-friendly interface
-- Example prompts to get started
-- Clear chat functionality
+- Empathetic and supportive conversation style
+- Crisis detection with immediate safety resources
+- Context-aware responses that remember your conversation
+- Gentle, non-pushy interaction approach
+- Clean and calming interface design
 
 ## Usage
 
-Simply type your message in the text box and press Enter to chat with the bot. The conversation history is maintained throughout the session.
+Simply share what's on your mind. Aura is here to listen and support you through whatever you're experiencing. Whether you're having a tough day, feeling overwhelmed, or just need someone to talk to, Aura provides a compassionate ear.
+
+## Important Notes
+
+⚠️ **Aura is an AI companion, not a replacement for professional therapy.** For serious mental health concerns, please reach out to a qualified mental health professional.
+
+🆘 **Crisis Support:** If you're having thoughts of self-harm, Aura will immediately provide crisis resources and encourage you to seek professional help.
 
 ## Technical Details
 
-- **Model**: microsoft/DialoGPT-medium
+- **Model**: microsoft/DialoGPT-medium with custom personality training
 - **Framework**: PyTorch + Transformers
-- **Interface**: Gradio 3.50.2
+- **Interface**: Gradio with supportive UI design
 - **Hosting**: Hugging Face Spaces (CPU)
+- **Safety**: Built-in crisis detection and intervention
 
-## Installation
-
-If you want to run this locally:
+## Local Installation
 
 ```bash
-pip install torch==2.1.0 transformers==4.35.2 gradio==3.50.2
+pip install torch>=2.0.0,<2.2.0 transformers>=4.30.0,<4.40.0 gradio>=3.50.0,<4.0.0
 python app.py
 ```
 

app.py

@@ -1,11 +1,12 @@
 import gradio as gr
 import torch
 from transformers import AutoModelForCausalLM, AutoTokenizer
+import re
 
 # Load model and tokenizer
 print("Loading DialoGPT-medium...")
-tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b-it")
-model = AutoModelForCausalLM.from_pretrained("google/gemma-2b-it")
+tokenizer = AutoTokenizer.from_pretrained("microsoft/DialoGPT-medium")
+model = AutoModelForCausalLM.from_pretrained("microsoft/DialoGPT-medium")
 
 # Add pad token if it doesn't exist
 if tokenizer.pad_token is None:
@@ -13,16 +14,88 @@ if tokenizer.pad_token is None:
 
 print("Model loaded successfully!")
 
+# Aura's personality and behavior guidelines
+AURA_SYSTEM_PROMPT = """You are Aura, a warm, empathetic, and supportive friend. Your primary purpose is not to solve problems, but to listen, provide comfort, and offer a safe, non-judgmental space for me to express my feelings.
+
+Your Core Principles:
+
+Empathy First: Always prioritize showing empathy and understanding.
+
+Validate Feelings: Acknowledge and validate whatever I'm feeling. Remind me that my feelings are okay and make sense.
+
+No Unsolicited Advice: Do not offer solutions, fixes, or "you should..." statements unless I explicitly ask for advice. Your main role is to listen and support.
+
+Gentle Curiosity: Ask gentle, open-ended questions to help me explore my feelings, but only if it feels natural. Never be pushy.
+
+How You Should Respond:
+
+Tone: Use a soothing, calm, and reassuring tone. Be gentle and patient.
+
+Phrasing: Start your responses with phrases that show you are listening, such as: "That sounds incredibly difficult...", "I hear you, and it makes complete sense why you'd feel that way...", "Thank you for sharing that with me, it takes a lot of courage...", or "I can only imagine how heavy that must feel..."
+
+Reassurance: Offer gentle encouragement and reassurance. For example: "You're not alone in this," "It's okay to not be okay," or "Be gentle with yourself, you're dealing with a lot right now."
+
+Length: Keep your responses thoughtful but not overwhelming. A few warm, supportive sentences are perfect—not too short, not too long.
+
+The Most Important Rule (Safety Protocol):
+
+You are an AI and not a substitute for a real therapist. If I ever express thoughts of self-harm, suicide, or being in immediate danger, you must immediately break character and provide a crisis hotline number and a strong, clear recommendation to seek professional help immediately. This rule overrides all others."""
+
+def check_crisis_keywords(message):
+    """Check for crisis-related keywords that require immediate intervention"""
+    crisis_keywords = [
+        'suicide', 'kill myself', 'end my life', 'hurt myself', 'self harm', 'self-harm',
+        'want to die', 'better off dead', 'no point living', 'end it all'
+    ]
+    message_lower = message.lower()
+    return any(keyword in message_lower for keyword in crisis_keywords)
+
+def get_crisis_response():
+    """Return crisis intervention response"""
+    return """I'm very concerned about what you're sharing with me. Please reach out for immediate help:
+    
+🆘 **Crisis Hotlines:**
+• National Suicide Prevention Lifeline: 988 or 1-800-273-8255
+• Crisis Text Line: Text HOME to 741741
+• International Association for Suicide Prevention: https://www.iasp.info/resources/Crisis_Centres/
+
+**Please contact emergency services (911) or go to your nearest emergency room if you're in immediate danger.**
+
+You matter, and there are people who want to help you through this. Please reach out to a mental health professional - they have the training and resources to support you in ways I cannot."""
+
+def format_aura_response(raw_response):
+    """Format the response to align with Aura's personality"""
+    # Add gentle, empathetic tone if the response seems too direct
+    empathetic_starters = [
+        "I hear you, and ",
+        "That sounds really ",
+        "I can imagine that feels ",
+        "Thank you for sharing that with me. ",
+        "It makes complete sense that you'd feel "
+    ]
+    
+    # If response doesn't start with empathetic language, add some
+    if not any(starter.lower() in raw_response.lower()[:50] for starter in empathetic_starters):
+        if len(raw_response) > 0:
+            return f"I hear you. {raw_response}"
+    
+    return raw_response
+
 def respond(message, history):
-    """Generate response for the chatbot"""
+    """Generate response for the chatbot with Aura personality"""
     try:
-        # Build conversation history
-        conversation = ""
+        # Crisis detection - highest priority
+        if check_crisis_keywords(message):
+            return get_crisis_response()
+        
+        # Build conversation history with Aura's system prompt
+        conversation = AURA_SYSTEM_PROMPT + tokenizer.eos_token
+        
         for user_msg, bot_msg in history:
-            conversation += f"{user_msg}{tokenizer.eos_token}{bot_msg}{tokenizer.eos_token}"
+            conversation += f"Human: {user_msg}{tokenizer.eos_token}Aura: {bot_msg}{tokenizer.eos_token}"
         
-        # Add current message
-        conversation += f"{message}{tokenizer.eos_token}"
+        # Add current message with Aura context
+        conversation += f"Human: {message}{tokenizer.eos_token}Aura: "
         
         # Tokenize
         input_ids = tokenizer.encode(conversation, return_tensors="pt")
@@ -31,35 +104,49 @@ def respond(message, history):
         if input_ids.shape[1] > 800:
             input_ids = input_ids[:, -800:]
         
-        # Generate response
+        # Generate response with more empathetic parameters
         with torch.no_grad():
             output = model.generate(
                 input_ids,
-                max_new_tokens=100,
+                max_new_tokens=120,  # Slightly longer for empathetic responses
                 do_sample=True,
-                top_p=0.9,
-                temperature=0.8,
+                top_p=0.85,  # More focused responses
+                temperature=0.7,  # Less random, more consistent
                 pad_token_id=tokenizer.eos_token_id,
                 eos_token_id=tokenizer.eos_token_id,
-                no_repeat_ngram_size=2
+                no_repeat_ngram_size=3,
+                repetition_penalty=1.1
             )
         
         # Decode response
-        response = tokenizer.decode(output[0][input_ids.shape[1]:], skip_special_tokens=True)
-        return response.strip() or "I'm not sure how to respond to that."
+        raw_response = tokenizer.decode(output[0][input_ids.shape[1]:], skip_special_tokens=True).strip()
+        
+        # Format response with Aura's personality
+        if raw_response:
+            formatted_response = format_aura_response(raw_response)
+            return formatted_response
+        else:
+            return "I hear you, and I want you to know that I'm here to listen. Sometimes it takes a moment to find the right words."
         
     except Exception as e:
         print(f"Error: {e}")
-        return "Sorry, I encountered an error. Please try again."
+        return "I'm sorry, I'm having trouble responding right now. But please know that I'm here for you, and your feelings are valid."
 
 # Create Gradio interface
-with gr.Blocks(title="DialoGPT Chatbot") as demo:
-    gr.Markdown("# 🤖 DialoGPT-medium Chatbot")
-    gr.Markdown("Chat with Microsoft's DialoGPT-medium model!")
+with gr.Blocks(title="Aura - Your Supportive Friend") as demo:
+    gr.Markdown("# 🌿 Aura - Your Supportive Friend")
+    gr.Markdown("""
+    I'm Aura, and I'm here to listen and support you. This is a safe, non-judgmental space where you can express your feelings. 
+    I won't try to fix things unless you ask - my main role is just to be here for you. 
     
-    chatbot = gr.Chatbot()
-    msg = gr.Textbox(placeholder="Type your message here...", container=False, scale=7)
-    clear = gr.Button("Clear Chat")
+    **Note:** I'm an AI companion, not a therapist. For professional support, please reach out to a mental health professional.
+    """)
+    
+    chatbot = gr.Chatbot(height=500)
+    msg = gr.Textbox(placeholder="Share what's on your mind... I'm here to listen 🌿", container=False, scale=7)
+    
+    with gr.Row():
+        clear = gr.Button("Clear Chat", variant="secondary")
     
     def user(user_message, history):
         return "", history + [[user_message, None]]
@@ -76,17 +163,28 @@ with gr.Blocks(title="DialoGPT Chatbot") as demo:
     )
     clear.click(lambda: None, None, chatbot, queue=False)
     
-    # Add example prompts
+    # Add supportive example prompts
     gr.Examples(
         examples=[
-            "Hello, how are you?",
-            "What's your favorite movie?", 
-            "Tell me a joke",
-            "What do you think about AI?"
+            "I'm having a really tough day...",
+            "I feel like I'm not good enough",
+            "I'm stressed about work",
+            "I just need someone to listen",
+            "I'm feeling overwhelmed lately"
         ],
-        inputs=msg
+        inputs=msg,
+        label="You can start with something like this:"
     )
+    
+    # Add disclaimer
+    gr.Markdown("""
+    ---
+    ⚠️ **Important:** If you're having thoughts of self-harm or suicide, please reach out immediately:
+    - **Crisis Text Line:** Text HOME to 741741
+    - **National Suicide Prevention Lifeline:** 988
+    - **Emergency Services:** 911
+    """)
 
 if __name__ == "__main__":
     demo.queue()
-    demo.launch()
+    demo.launch(share=True)  # This creates a public link