Upload 5 files

app.py

@@ -0,0 +1,47 @@
+import os
+import shutil
+from flask import Flask, render_template, request, jsonify
+from transformers import AutoModelForCausalLM, AutoTokenizer
+import torch
+
+app = Flask(__name__)
+
+# Kill any process using port 7860
+def kill_port(port):
+    for proc in psutil.process_iter(attrs=["pid", "connections"]):
+        for conn in proc.info["connections"]:
+            if conn.laddr.port == port:
+                os.kill(proc.info["pid"], 9)
+
+kill_port(7860)  # Ensure Flask doesn't crash due to a used port
+
+# Define cache directory
+cache_dir = os.path.expanduser("~/.cache/huggingface")
+
+# Load Myanmarsar-GPT (1.42B params) from Hugging Face
+MODEL_NAME = "simbolo-ai/Myanmarsar-GPT"
+tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME, cache_dir=cache_dir)
+model = AutoModelForCausalLM.from_pretrained(MODEL_NAME, cache_dir=cache_dir)
+
+# Function to generate chatbot responses
+def generate_response(prompt):
+    inputs = tokenizer(prompt, return_tensors="pt")
+    with torch.no_grad():
+        output = model.generate(**inputs, max_length=200)
+    return tokenizer.decode(output[0], skip_special_tokens=True)
+
+# Serve the HTML page
+@app.route("/")
+def home():
+    return render_template("index.html")
+
+# API route for chatbot responses
+@app.route("/chat", methods=["POST"])
+def chat():
+    user_message = request.json.get("message", "")
+    bot_reply = generate_response(user_message)
+    return jsonify({"reply": bot_reply})
+
+if __name__ == "__main__":
+    port = int(os.environ.get("PORT", 7860))  # Default to 7860, but use any assigned port
+    app.run(host="0.0.0.0", port=port)

requirements.txt

@@ -0,0 +1,5 @@
+flask
+torch
+transformers==4.37.0
+accelerate
+psutil

static/css/style.css

@@ -0,0 +1,43 @@
+/* static/css/style.css */
+body {
+    font-family: Arial, sans-serif;
+    display: flex;
+    justify-content: center;
+    align-items: center;
+    height: 100vh;
+    background-color: #f5f5f5;
+}
+.chat-container {
+    width: 90%;
+    max-width: 400px;
+    background: white;
+    border-radius: 10px;
+    box-shadow: 0px 0px 10px rgba(0, 0, 0, 0.1);
+    overflow: hidden;
+}
+.chat-box {
+    height: 400px;
+    overflow-y: auto;
+    padding: 10px;
+    border-bottom: 1px solid #ccc;
+}
+.input-area {
+    display: flex;
+    padding: 10px;
+    background: #eee;
+}
+input {
+    flex: 1;
+    padding: 8px;
+    border: none;
+    border-radius: 5px;
+}
+button {
+    margin-left: 5px;
+    padding: 8px 10px;
+    background: blue;
+    color: white;
+    border: none;
+    border-radius: 5px;
+    cursor: pointer;
+}

static/js/script.js

@@ -0,0 +1,44 @@
+document.getElementById("send-button").addEventListener("click", function() {
+    sendMessage();
+});
+
+document.getElementById("user-input").addEventListener("keypress", function(event) {
+    if (event.key === "Enter") {
+        sendMessage();
+    }
+});
+
+const micButton = document.getElementById("mic-button");
+const recognition = new (window.SpeechRecognition || window.webkitSpeechRecognition)();
+recognition.lang = "my-MM"; // Set to Burmese
+recognition.continuous = false;
+
+micButton.addEventListener("click", () => {
+    recognition.start();
+});
+
+recognition.onresult = (event) => {
+    document.getElementById("user-input").value = event.results[0][0].transcript;
+};
+
+function sendMessage() {
+    let userMessage = document.getElementById("user-input").value;
+    let chatbox = document.getElementById("chatbox");
+
+    if (userMessage.trim() === "") return;
+
+    // Display user message
+    chatbox.innerHTML += `<p><strong>You:</strong> ${userMessage}</p>`;
+    document.getElementById("user-input").value = "";
+
+    fetch("/chat", {
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        body: JSON.stringify({ message: userMessage })
+    })
+    .then(response => response.json())
+    .then(data => {
+        chatbox.innerHTML += `<p><strong>Bot:</strong> ${data.reply}</p>`;
+    })
+    .catch(error => console.error("Error:", error));
+}

templates/index.html

@@ -0,0 +1,19 @@
+<html lang="en">
+<head>
+    <meta charset="UTF-8">
+    <meta name="viewport" content="width=device-width, initial-scale=1.0">
+    <title>Chatbot UI</title>
+    <link rel="stylesheet" href="{{ url_for('static', filename='css/style.css') }}">
+</head>
+<body>
+    <div class="chat-container">
+        <div class="chat-box" id="chatbox"></div>
+        <div class="input-area">
+            <button id="mic-button">🎤</button>
+            <input type="text" id="user-input" placeholder="Type a message...">
+            <button id="send-button">Send</button>
+        </div>
+    </div>
+    <script src="{{ url_for('static', filename='js/script.js') }}"></script>
+</body>
+</html>