Initial commit without background.jpg

.gitattributes

@@ -0,0 +1,39 @@
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz filter=lfs diff=lfs merge=lfs -text
+*.zip filter=lfs diff=lfs merge=lfs -text
+*.zst filter=lfs diff=lfs merge=lfs -text
+*tfevents* filter=lfs diff=lfs merge=lfs -text
+<<<<<<< HEAD
+static/background.jpg filter=lfs diff=lfs merge=lfs -text
+=======
+>>>>>>> d81314b530b40e15259c3fcc60df37c5f08161cb

Dockerfile

@@ -0,0 +1,20 @@
+# Use an official Python runtime as a parent image
+FROM python:3.11-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the current directory contents into the container at /app
+COPY . /app
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r Requirements.txt
+
+# Make port 8080 available to the world outside this container
+EXPOSE 8080
+
+# Define environment variable for GCP
+ENV PORT=8080
+
+# Run the application using Gunicorn
+CMD ["gunicorn", "--bind", "0.0.0.0:8080", "application:application"]

README.md

@@ -0,0 +1,11 @@
+---
+title: HarmonyBot
+emoji: 👀
+colorFrom: green
+colorTo: pink
+sdk: docker
+pinned: false
+license: apache-2.0
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

Requirements.txt

@@ -0,0 +1,7 @@
+Flask
+gunicorn
+huggingface_hub
+datasets
+markdown2
+requests
+transformers

application.py

@@ -0,0 +1,99 @@
+import os
+from flask import Flask, request, jsonify, render_template, send_from_directory
+from huggingface_hub import InferenceClient
+from datasets import load_dataset
+import markdown2
+import signal
+
+application = Flask(__name__, static_folder='static', template_folder='templates')
+
+hf_token = os.getenv("HF_TOKEN")
+
+chat_doctor_dataset = load_dataset("avaliev/chat_doctor")
+mental_health_dataset = load_dataset("Amod/mental_health_counseling_conversations")
+
+client = InferenceClient(
+    "meta-llama/Meta-Llama-3-8B-Instruct",
+    token=hf_token,
+)
+
+def select_relevant_context(user_input):
+    mental_health_keywords = [
+        "anxious", "depressed", "stress", "mental health", "counseling", 
+        "therapy", "feelings", "worthless", "suicidal", "panic", "anxiety"
+    ]
+    medical_keywords = [
+        "symptoms", "diagnosis", "treatment", "doctor", "prescription", "medication",
+        "pain", "illness", "disease", "infection", "surgery"
+    ]
+
+    # Check if the input contains any mental health-related keywords
+    if any(keyword in user_input.lower() for keyword in mental_health_keywords):
+        example = mental_health_dataset['train'][0]
+        context = f"Counselor: {example['Response']}\nUser: {example['Context']}"
+    # Check if the input contains any medical-related keywords
+    elif any(keyword in user_input.lower() for keyword in medical_keywords):
+        example = chat_doctor_dataset['train'][0]
+        context = f"Doctor: {example['input']}\nPatient: {example['output']}"
+    else:
+        # If no specific keywords are found, provide a general response
+        context = "You are a general assistant. Respond to the user's query in a helpful manner."
+
+    return context
+
+def create_prompt(context, user_input):
+    prompt = (
+        f"{context}\n\n"
+        f"User: {user_input}\nAssistant:"
+    )
+    return prompt
+
+# Function to render Markdown into HTML
+def render_markdown(text):
+    return markdown2.markdown(text)
+
+@application.route('/')
+def index():
+    return render_template('index.html')
+
+@application.route('/static/<path:path>')
+def send_static(path):
+    return send_from_directory('static', path)
+
+@application.route('/chat', methods=['POST'])
+def chat():
+    user_input = request.json['message']
+    
+    context = select_relevant_context(user_input)
+    
+    prompt = create_prompt(context, user_input)
+    
+    response = ""
+    for message in client.chat_completion(
+        messages=[{"role": "user", "content": prompt}],
+        max_tokens=500,
+        stream=True,
+    ):
+        response += message.choices[0].delta.content
+    
+    formatted_response = render_markdown(response)
+    
+    return jsonify({"response": formatted_response})
+
+@application.route('/shutdown', methods=['POST'])
+def shutdown():
+    if request.environ.get('werkzeug.server.shutdown'):
+        shutdown_server()
+    else:
+        os.kill(os.getpid(), signal.SIGINT)
+    return jsonify({"message": "Server is shutting down..."})
+
+def shutdown_server():
+    func = request.environ.get('werkzeug.server.shutdown')
+    if func is None:
+        os.kill(os.getpid(), signal.SIGINT)  # Kill the process if Werkzeug is not available
+    else:
+        func()
+
+if __name__ == '__main__':
+    application.run(debug=False)

gitignore

@@ -0,0 +1 @@
+static/background.jpg

static/script.js

@@ -0,0 +1,49 @@
+async function sendMessage() {
+    const userInput = document.getElementById('user-input').value;
+    if (!userInput) return;
+
+    const chatBox = document.getElementById('chat-box');
+
+    // Display user's message
+    const userMessage = document.createElement('div');
+    userMessage.classList.add('message', 'user');
+    userMessage.textContent = userInput;
+    chatBox.appendChild(userMessage);
+
+    // Clear the input field
+    document.getElementById('user-input').value = '';
+
+    // Send the message to the server
+    const response = await fetch('/chat', {
+        method: 'POST',
+        headers: { 'Content-Type': 'application/json' },
+        body: JSON.stringify({ message: userInput })
+    });
+
+    const data = await response.json();
+
+    // Create a new div for the bot's response
+    const botMessage = document.createElement('div');
+    botMessage.classList.add('message', 'bot');
+    botMessage.innerHTML = data.response; // Use innerHTML to render Markdown
+
+    chatBox.appendChild(botMessage);
+
+    // Scroll to the bottom of the chat box
+    chatBox.scrollTop = chatBox.scrollHeight;
+}
+
+function handleKeyPress(event) {
+    if (event.key === 'Enter') {
+        sendMessage();
+    }
+}
+
+async function exitChat() {
+    await fetch('/shutdown', { method: 'POST' });
+}
+
+function newChat() {
+    const chatBox = document.getElementById('chat-box');
+    chatBox.innerHTML = '<div class="message bot">Hi, Welcome to the Medical Bot. What is your query?</div>';
+}

static/style.css

@@ -0,0 +1,154 @@
+body {
+    font-family: 'Helvetica Neue', Arial, sans-serif;
+    margin: 0;
+    padding: 0;
+    background: url('/static/background.jpg') no-repeat center center fixed;
+    background-size: cover;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    height: 100vh;
+}
+
+.chat-container {
+    width: 100%;
+    max-width: 900px;
+    height: 90vh;
+    display: flex;
+    background-color: rgb(236, 59, 177);
+    box-shadow: 0 0 20px rgba(0, 0, 0, 0.1);
+    border-radius: 10px;
+    overflow: hidden;
+}
+
+.sidebar {
+    width: 120px;
+    background-color: hsl(280, 48%, 28%);
+    display: flex;
+    flex-direction: column;
+    align-items: center;
+    padding: 20px 10px;
+}
+
+.sidebar button {
+    background-color: #071c32;
+    color: rgb(255, 255, 255);
+    border: none;
+    padding: 15px 20px;
+    border-radius: 20px;
+    margin: 20px 0;
+    cursor: pointer;
+    font-size: 16px;
+    box-shadow: 0 5px 10px rgba(0, 0, 0, 0.2);
+    transition: transform 0.3s ease, box-shadow 0.3s ease;
+    width: 100px; /* Ensure buttons are the same width */
+    text-align: center;
+}
+
+.sidebar button:hover {
+    transform: scale(1.05);
+    box-shadow: 0 8px 15px rgba(0, 0, 0, 0.3);
+}
+
+.main-content {
+    flex: 1;
+    display: flex;
+    flex-direction: column;
+    background-color: rgba(125, 130, 131, 0.455);
+}
+
+.chat-box {
+    flex: 1;
+    padding: 20px;
+    overflow-y: auto;
+    border-bottom: 1px solid #e91199;
+}
+
+.message {
+    margin: 10px 0;
+    padding: 10px 15px;
+    border-radius: 20px;
+    max-width: 75%;
+    word-wrap: break-word;
+    box-shadow: 0 0 5px rgb(203, 60, 146);
+}
+
+/* Markdown specific styling */
+.message.bot p {
+    margin: 0 0 10px;
+}
+
+.message.bot ul {
+    margin: 10px 0;
+    padding-left: 20px;
+}
+
+.message.bot ol {
+    margin: 10px 0;
+    padding-left: 20px;
+}
+
+.message.bot li {
+    margin-bottom: 5px;
+}
+
+.message.bot code {
+    background-color: rgba(27, 31, 35, 0.05);
+    padding: 2px 4px;
+    border-radius: 3px;
+    font-family: 'Courier New', Courier, monospace;
+}
+
+.message.bot pre {
+    background-color: rgba(27, 31, 35, 0.1);
+    padding: 10px;
+    border-radius: 5px;
+    overflow-x: auto;
+    font-family: 'Courier New', Courier, monospace;
+}
+
+/* User messages */
+.message.user {
+    background-color: hwb(285 20% 63% / 0.815);
+    color: rgba(214, 199, 199, 0.879);
+    align-self: flex-end;
+}
+
+.message.bot {
+    background-color: hwb(297 45% 14% / 0.815) ;
+    color: #151415;
+    align-self: flex-start;
+}
+
+.input-container {
+    display: flex;
+    padding: 15px;
+    border-top: 1px solid #ddd;
+    background-color: rgba(250, 250, 250, 0.9);
+}
+
+input[type="text"] {
+    flex: 1;
+    padding: 10px;
+    border: 1px solid #ddd;
+    border-radius: 20px;
+    margin-right: 10px;
+    font-size: 16px;
+}
+
+button.send-button {
+    padding: 10px 20px;
+    background-color: hsl(280, 48%, 28%);
+    color: white;
+    border: none;
+    border-radius: 20px;
+    cursor: pointer;
+    font-size: 16px;
+    box-shadow: 0 5px 10px rgba(0, 0, 0, 0.2);
+    transition: transform 0.3s ease, box-shadow 0.3s ease;
+}
+
+button.send-button:hover {
+    transform: scale(1.05);
+    box-shadow: 0 8px 15px rgba(0, 0, 0, 0.3);
+}

templates/index.html

@@ -0,0 +1,28 @@
+<!DOCTYPE html>
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Medical Bot</title>
+    <link rel="stylesheet" href="static/style.css">
+</head>
+<body>
+    <div class="chat-container">
+        <div class="sidebar">
+            <button class="new-chat-button" onclick="newChat()">New</button>
+            <button class="exit-button" onclick="exitChat()">Exit</button>
+        </div>
+        <div class="main-content">
+            <div id="chat-box" class="chat-box">
+                <div class="message bot">Hi, Welcome to the Medical Bot. What is your query?</div>
+            </div>
+            <div class="input-container">
+                <input type="text" id="user-input" placeholder="Type your message here..." onkeypress="handleKeyPress(event)" />
+                <button class="send-button" onclick="sendMessage()">Send</button>
+            </div>
+        </div>
+    </div>
+
+    <script src="static/script.js"></script>
+</body>
+</html>