Spaces:

Gopikanth123
/

voicechat

Sleeping

App Files Files Community

Gopikanth123 commited on 25 days ago

Commit

82bfd91

verified ·

1 Parent(s): 7c59f65

Update main.py

Browse files

Files changed (1) hide show

main.py +145 -474

main.py CHANGED Viewed

@@ -1,480 +1,151 @@
-<!DOCTYPE html>
-<html lang="en">
-<head>
-    <meta charset="UTF-8">
-    <meta name="viewport" content="width=device-width, initial-scale=1.0">
-    <title>TAJ HOTEL CHATBOT</title>
-    <link href="https://fonts.googleapis.com/css2?family=Roboto:wght@400;500;700&display=swap" rel="stylesheet">
-    <style>
-        :root {
-            --primary-color: #007bff; /* Default blue */
-            --bot-message-bg: #f0f2f5; /* Light Gray */
-            --user-message-bg: #007bff; /* User Blue */
-            --user-message-text: #fff; /* User Message White */
-            --accent-color: #6a0dad; /* Default Purple */
-            --background-color: #fff; /* Default Background White */
-            --shadow-color: rgba(0, 0, 0, 0.2); /* Default Shadow */
-            --input-bg: #f0f2f5; /* Default Input Background */
-            --input-border: #ccc; /* Default Input Border */
-        }
-        /* Themes */
-        .theme-calm-azure {
-            --background-color: #E3F2FD;
-            --bot-message-bg: #BBDEFB;
-            --user-message-bg: #2196F3;
-            --user-message-text: #FFFFFF;
-            --input-bg: #FFFFFF;
-            --input-border: #BDBDBD;
-            --accent-color: #1976D2;
-        }
-        .theme-elegant-charcoal {
-            --background-color: #263238;
-            --bot-message-bg: #37474F;
-            --user-message-bg: #FF5722;
-            --user-message-text: #FFFFFF;
-            --input-bg: #455A64;
-            --input-border: #CFD8DC;
-            --accent-color: #FF9800;
-        }
-        .theme-fresh-greenery {
-            --background-color: #E8F5E9;
-            --bot-message-bg: #C8E6C9;
-            --user-message-bg: #4CAF50;
-            --user-message-text: #FFFFFF;
-            --input-bg: #FFFFFF;
-            --input-border: #A5D6A7;
-            --accent-color: #388E3C;
-        }
-        .theme-soft-lavender {
-            --background-color: #F3E5F5;
-            --bot-message-bg: #E1BEE7;
-            --user-message-bg: #9C27B0;
-            --user-message-text: #FFFFFF;
-            --input-bg: #FFFFFF;
-            --input-border: #D1C4E9;
-            --accent-color: #7B1FA2;
-        }
-        .theme-bright-summer {
-            --background-color: #FFEB3B;
-            --bot-message-bg: #FFF9C4;
-            --user-message-bg: #F44336;
-            --user-message-text: #FFFFFF;
-            --input-bg: #FFFFFF;
-            --input-border: #FBC02D;
-            --accent-color: #C62828;
-        }
-        * {
-            margin: 0;
-            padding: 0;
-            box-sizing: border-box;
-        }
-        body {
-            font-family: 'Roboto', sans-serif;
-            background-color: var(--background-color);
-            transition: background 0.5s ease;
-            overflow: hidden; /* Prevent scroll on body */
-        }
-        .container {
-            display: flex;
-            flex-direction: column;
-            height: 100vh;
-            width: 100%;
-            overflow: hidden; /* Prevent scroll on container */
-        }
-        /* Header */
-        .header {
-            background: var(--accent-color);
-            color: #fff;
-            font-size: 1.5rem;
-            font-weight: 700;
-            text-align: center;
-            padding: 20px;
-            box-shadow: 0 4px 8px var(--shadow-color);
-            position: relative;
-            z-index: 1000;
-        }
-        /* Chatbox */
-        .chat-box {
-            flex: 1;
-            display: flex;
-            flex-direction: column;
-            overflow-y: auto;
-            padding: 20px;
-            background: linear-gradient(to bottom right, #f5f7fa, #c3cfe2);
-            border-radius: 10px;
-            margin: 20px;
-            box-shadow: 0 4px 8px var(--shadow-color);
-            position: relative;
-            z-index: 10;
-        }
-        .message {
-            max-width: 75%;
-            padding: 12px 18px;
-            border-radius: 20px;
-            box-shadow: 0 3px 6px var(--shadow-color);
-            margin-bottom: 10px;
-            opacity: 0;
-            animation: fadeIn 0.3s forwards; /* Changed to forwards for delay effect */
-        }
-        .user-message {
-            align-self: flex-end;
-            background: var(--user-message-bg);
-            color: var(--user-message-text);
-            border-radius: 15px 20px 20px 20px;
-            animation: slideInRight 0.5s forwards; /* Sliding effect on user message */
-        }
-        .bot-message {
-            align-self: flex-start;
-            background: var(--bot-message-bg);
-            color: #333;
-            border-radius: 20px 15px 20px 20px;
-            animation: slideInLeft 0.5s forwards; /* Sliding effect on bot message */
-        }
-        /* Footer */
-        .footer {
-            background: #ffffff;
-            padding: 15px;
-            display: flex;
-            justify-content: center;
-            align-items: center;
-            box-shadow: 0 -4px 8px var(--shadow-color);
-            position: relative;
-            z-index: 1000;
-        }
-        .footer input[type="text"] {
-            width: 75%;
-            padding: 15px;
-            border: 1px solid var(--input-border);
-            border-radius: 20px;
-            margin-right: 10px;
-            box-shadow: 0 2px 4px var(--shadow-color);
-            transition: border 0.3s, box-shadow 0.3s; /* Added shadow transition */
-            background-color: var(--input-bg);
-            outline: none;
-        }
-        .footer input[type="text"]:focus {
-            border-color: var(--accent-color);
-            box-shadow: 0 0 10px var(--accent-color);
-        }
-        button {
-            background: var(--accent-color);
-            color: #fff;
-            border: none;
-            padding: 10px 20px;
-            border-radius: 20px;
-            font-size: 1rem;
-            cursor: pointer;
-            box-shadow: 0 4px 10px var(--shadow-color);
-            transition: background 0.3s ease, transform 0.2s ease, box-shadow 0.2s ease;
-        }
-        button:hover {
-            background: #4b0082;
-            transform: scale(1.05);
-            box-shadow: 0 4px 20px rgba(0, 0, 0, 0.5); /* Shadow effect on hover */
-        }
-        /* Settings */
-        .settings {
-            display: flex;
-            justify-content: space-between;
-            align-items: center;
-            padding: 10px;
-            background: #f0f2f5;
-            box-shadow: 0 2px 5px var(--shadow-color);
-            position: relative;
-            z-index: 1000;
-        }
-        .color-picker,
-        .theme-toggle {
-            display: flex;
-            align-items: center;
-            position:relative;
-        }
-        .color-circle {
-            width: 20px;
-            height: 20px;
-            border-radius: 50%;
-            margin: 0 5px;
-            cursor: pointer;
-            box-shadow: 0 2px 5px var(--shadow-color);
-            transition: transform 0.3s; /* Adding circle hover effect */
-        }
-        .color-circle:hover {
-            transform: scale(1.2); /* Scale up on hover */
-        }
-        /* Animations */
-        @keyframes fadeIn {
-            from {
-                opacity: 0;
-                transform: translateY(20px);
-            }
-            to {
-                opacity: 1;
-                transform: translateY(0);
-            }
-        }
-        @keyframes slideInRight {
-            from {
-                opacity: 0;
-                transform: translateX(100%);
-            }
-            to {
-                opacity: 1;
-                transform: translateX(0);
-            }
-        }
-        @keyframes slideInLeft {
-            from {
-                opacity: 0;
-                transform: translateX(-100%);
-            }
-            to {
-                opacity: 1;
-                transform: translateX(0);
-            }
-        }
-        /* Background animation */
-        @keyframes backgroundAnimate {
-            0% {
-                background-color: rgba(255, 255, 255, 0.05);
-            }
-            50% {
-                background-color: rgba(255, 255, 255, 0.1);
-            }
-            100% {
-                background-color: rgba(255, 255, 255, 0.05);
-            }
-        }
-        /* VFX related styles */
-        .vfx {
-            position: absolute;
-            top: 0;
-            left: 0;
-            width: 100%;
-            height: 100%;
-            pointer-events: none;
-            animation: backgroundAnimate 5s infinite; /* Continuously animate background */
-            z-index: 0; /* Background layer */
-        }
-    </style>
-</head>
-<body>
-    <div class="vfx"></div> <!-- Background effects -->
-    <div class="container">
-        <!-- Settings -->
-        <div class="settings">
-            <div class="theme-toggle">
-                <label for="theme-select">Select Theme:</label>
-                <select id="theme-select">
-                    <option value="default">Default</option>
-                    <option value="calm-azure">Calm Azure</option>
-                    <option value="elegant-charcoal">Elegant Charcoal</option>
-                    <option value="fresh-greenery">Fresh Greenery</option>
-                    <option value="soft-lavender">Soft Lavender</option>
-                    <option value="bright-summer">Bright Summer</option>
-                </select>
-            </div>
-            <div class="color-picker">
-                <label>Accent Color:</label>
-                <div class="color-circle" style="background-color: #6a0dad;" onclick="changeColor('#6a0dad')"></div>
-                <div class="color-circle" style="background-color: #ff4500;" onclick="changeColor('#ff4500')"></div>
-                <div class="color-circle" style="background-color: #007bff;" onclick="changeColor('#007bff')"></div>
-                <div class="color-circle" style="background-color: #28a745;" onclick="changeColor('#28a745')"></div>
-            </div>
-        </div>
-        <!-- Header   -->
-        <div class="header">TAJ HOTEL CHATBOT</div>
-        <!-- Chatbox -->
-        <div class="chat-box" id="chat-box"></div>
-        <!-- Footer -->
-        <div class="footer">
-          <input type="text" id="user-input" placeholder="Type your message..." />
-          <button id="send-btn">Send</button>
-          <button id="voice-btn">🎤 Start Voice Input</button>
-          <!-- Language dropdown -->
-          <select id="language-select">
-              <option value="english" selected>English</option>
-              <option value="telugu">Telugu</option>
-              <option value="hindi">Hindi</option>
-          </select>
-        </div>
-    </div>
-    <script>
-        const chatBox = document.getElementById('chat-box');
-        const voiceBtn = document.getElementById('voice-btn');
-        const sendBtn = document.getElementById('send-btn');
-        const userInput = document.getElementById('user-input');
-        const themeSelect = document.getElementById('theme-select');
-        const languageSelect = document.getElementById('language-select');
-        // Add message to chatbox with visual effects
-        function addMessage(sender, text) {
-            const msgDiv = document.createElement('div');
-            msgDiv.classList.add('message', sender);
-            msgDiv.textContent = text;
-            chatBox.appendChild(msgDiv);
-            chatBox.scrollTop = chatBox.scrollHeight;
-        }
-        // Speech Recognition Setup
-        const recognition = new (window.SpeechRecognition || window.webkitSpeechRecognition)();
-        // recognition.lang = 'en-US';
-        // Function to set language for speech recognition
-        function setRecognitionLanguage() {
-            const selectedLanguage = languageSelect.value;
-            switch (selectedLanguage) {
-                case 'telugu':
-                    recognition.lang = 'te-IN'; // Telugu
-                    break;
-                case 'hindi':
-                    recognition.lang = 'hi-IN'; // Hindi
-                    break;
-                default:
-                    recognition.lang = 'en-US'; // English
-                    break;
-            }
-        }
-        // voiceBtn.addEventListener('click', () => recognition.start());
-        voiceBtn.addEventListener('click', () => {
-            setRecognitionLanguage();  // Set the language before starting speech recognition
-            recognition.start();
-        });
-        recognition.addEventListener('result', (e) => {
-            const transcript = e.results[0][0].transcript;
-            addMessage('user-message', transcript);
-            sendUserMessage(transcript);
-        });
-        // Function to change the accent color
-        function changeColor(color) {
-            document.documentElement.style.setProperty('--accent-color', color);
-        }
-        // Function to change the theme
-        function changeTheme(theme) {
-            document.documentElement.classList.remove('theme-calm-azure', 'theme-elegant-charcoal', 'theme-fresh-greenery', 'theme-soft-lavender', 'theme-bright-summer');
-            if (theme !== 'default') {
-                document.documentElement.classList.add('theme-' + theme);
-            }
-        }
-        // Function to send user input and selected language to backend
-        function sendUserMessage(message) {
-            const selectedLanguage = languageSelect.value; // Get the selected language
-            // Send the message and selected language to the backend
-            fetch('/chat', {
-                method: 'POST',
-                headers: {
-                    'Content-Type': 'application/json',
-                },
-                body: JSON.stringify({
-                    message: message,
-                    language: selectedLanguage, // Include the selected language in the request body
-                }),
-            })
-            .then(response => response.json())
-            .then(data => {
-                const botResponse = data.response;
-                addMessage('bot-message', botResponse);
-                speakResponse(botResponse);
-            })
-            .catch(error => {
-                console.error("Error:", error);
-                addMessage('bot-message', "Sorry, I couldn't process that.");
-            });
-        }
-        // // Text-to-Speech Function
-        // function speakResponse(text) {
-        //     const utterance = new SpeechSynthesisUtterance(text);
-        //     utterance.lang = 'en-US';
-        //     window.speechSynthesis.speak(utterance);
-        // }
-        // Text-to-Speech Function
-        function speakResponse(text) {
-            const utterance = new SpeechSynthesisUtterance(text);
-            // Set the language for text-to-speech based on the selected language
-            const selectedLanguage = languageSelect.value;
-            switch (selectedLanguage) {
-                case 'telugu':
-                    utterance.lang = 'te-IN'; // Telugu
-                    break;
-                case 'hindi':
-                    utterance.lang = 'hi-IN'; // Hindi
-                    break;
-                default:
-                    utterance.lang = 'en-US'; // English
-                    break;
-            }
-            window.speechSynthesis.speak(utterance);
-        }
-        // Event listeners for buttons
-        sendBtn.addEventListener('click', () => {
-            const message = userInput.value.trim();
-            if (message) {
-                addMessage('user-message', message);
-                sendUserMessage(message);
-                userInput.value = ''; // Clear input field after sending
-            }
-        });
-        // Handle pressing 'Enter' key for sending messages
-        userInput.addEventListener('keypress', (e) => {
-            if (e.key === 'Enter') {
-                sendBtn.click(); // Trigger button click
-            }
-        });
-        // Update theme when selected from dropdown
-        themeSelect.addEventListener('change', (e) => {
-            changeTheme(e.target.value);
-        });
-        recognition.addEventListener('error', (event) => {
-            console.error("Speech recognition error", event);
-        });
-    </script>
-  </body>
-</html>

+import os
+import shutil
+from flask import Flask, render_template, request, jsonify
+from llama_index.core import StorageContext, load_index_from_storage, VectorStoreIndex, SimpleDirectoryReader, ChatPromptTemplate, Settings
+from llama_index.llms.huggingface import HuggingFaceInferenceAPI
+from llama_index.embeddings.huggingface import HuggingFaceEmbedding
+from huggingface_hub import InferenceClient
+from transformers import AutoTokenizer, AutoModel
+# Ensure HF_TOKEN is set
+HF_TOKEN = os.getenv("HF_TOKEN")
+if not HF_TOKEN:
+    raise ValueError("HF_TOKEN environment variable not set.")
+repo_id = "meta-llama/Meta-Llama-3-8B-Instruct"
+llm_client = InferenceClient(
+    model=repo_id,
+    token=HF_TOKEN,
+)
+# Configure Llama index settings
+Settings.llm = HuggingFaceInferenceAPI(
+    model_name=repo_id,
+    tokenizer_name=repo_id,
+    context_window=3000,
+    token=HF_TOKEN,
+    max_new_tokens=512,
+    generate_kwargs={"temperature": 0.1},
+)
+# Settings.embed_model = HuggingFaceEmbedding(
+#     model_name="BAAI/bge-small-en-v1.5"
+# )
+# Replace the embedding model with XLM-R
+Settings.embed_model = HuggingFaceEmbedding(
+    model_name="xlm-roberta-base"  # XLM-RoBERTa model for multilingual support
+)
+# Configure tokenizer and model if required
+tokenizer = AutoTokenizer.from_pretrained("xlm-roberta-base")
+model = AutoModel.from_pretrained("xlm-roberta-base")
+PERSIST_DIR = "db"
+PDF_DIRECTORY = 'data'
+# Ensure directories exist
+os.makedirs(PDF_DIRECTORY, exist_ok=True)
+os.makedirs(PERSIST_DIR, exist_ok=True)
+chat_history = []
+current_chat_history = []
+def data_ingestion_from_directory():
+    # Clear previous data by removing the persist directory
+    if os.path.exists(PERSIST_DIR):
+        shutil.rmtree(PERSIST_DIR)  # Remove the persist directory and all its contents
+    # Recreate the persist directory after removal
+    os.makedirs(PERSIST_DIR, exist_ok=True)
+    # Load new documents from the directory
+    new_documents = SimpleDirectoryReader(PDF_DIRECTORY).load_data()
+    # Create a new index with the new documents
+    index = VectorStoreIndex.from_documents(new_documents)
+    # Persist the new index
+    index.storage_context.persist(persist_dir=PERSIST_DIR)
+def handle_query(query):
+    context_str = ""
+    # Build context from current chat history
+    for past_query, response in reversed(current_chat_history):
+        if past_query.strip():
+            context_str += f"User asked: '{past_query}'\nBot answered: '{response}'\n"
+    chat_text_qa_msgs = [
+        (
+            "user",
+            """
+            You are the Hotel voice chatbot and your name is hotel helper. Your goal is to provide accurate, professional, and helpful answers to user queries based on the hotel's data. Always ensure your responses are clear and concise. Give response within 10-15 words only. You need to give an answer in the same language used by the user.
+            {context_str}
+            Question:
+            {query_str}
+            """
+        )
+    ]
+    text_qa_template = ChatPromptTemplate.from_messages(chat_text_qa_msgs)
+    storage_context = StorageContext.from_defaults(persist_dir=PERSIST_DIR)
+    index = load_index_from_storage(storage_context)
+    # context_str = ""
+    # # Build context from current chat history
+    # for past_query, response in reversed(current_chat_history):
+    #     if past_query.strip():
+    #         context_str += f"User asked: '{past_query}'\nBot answered: '{response}'\n"
+    query_engine = index.as_query_engine(text_qa_template=text_qa_template, context_str=context_str)
+    print(f"Querying: {query}")
+    answer = query_engine.query(query)
+    # Extracting the response
+    if hasattr(answer, 'response'):
+        response = answer.response
+    elif isinstance(answer, dict) and 'response' in answer:
+        response = answer['response']
+    else:
+        response = "I'm sorry, I couldn't find an answer to that."
+    # Append to chat history
+    current_chat_history.append((query, response))
+    return response
+app = Flask(__name__)
+# Data ingestion
+data_ingestion_from_directory()
+# Generate Response
+def generate_response(query):
+    try:
+        # Call the handle_query function to get the response
+        bot_response = handle_query(query)
+        return bot_response
+    except Exception as e:
+        return f"Error fetching the response: {str(e)}"
+# Route for the homepage
+@app.route('/')
+def index():
+    return render_template('index.html')
+# Route to handle chatbot messages
+@app.route('/chat', methods=['POST'])
+def chat():
+    try:
+        user_message = request.json.get("message")
+        if not user_message:
+            return jsonify({"response": "Please say something!"})
+        bot_response = generate_response(user_message)
+        return jsonify({"response": bot_response})
+    except Exception as e:
+        return jsonify({"response": f"An error occurred: {str(e)}"})
+if __name__ == '__main__':
+    app.run(debug=True)