Upload 5 files

Dockerfile

@@ -0,0 +1,33 @@
+# Use an official Python runtime as the parent image
+FROM python:3.9-slim
+
+# Set the working directory in the container
+WORKDIR /app
+
+# Copy the Python requirements file into the container
+COPY requirements.txt .
+
+# Install any needed packages specified in requirements.txt
+RUN pip install --no-cache-dir -r requirements.txt
+
+# Copy the backend Python script
+COPY app.py .
+
+# Create directories for templates and static files
+RUN mkdir templates static
+
+# Copy the HTML template
+COPY templates/index.html templates/
+
+# Copy the JavaScript file
+COPY static/script.js static/
+
+# Make port 8000 available to the world outside this container
+EXPOSE 8000
+
+# Define environment variable
+ENV FLASK_APP=app.py
+ENV FLASK_RUN_HOST=0.0.0.0
+
+# Run app.py when the container launches
+CMD ["flask", "run", "--port=8000"]

app.py

@@ -0,0 +1,83 @@
+import os
+from flask import Flask, request, jsonify, render_template
+from transformers import pipeline
+import mysql.connector
+from groq import Groq
+
+app = Flask(name)
+
+# Initialize the text generation pipeline
+pipe = pipeline("text-generation", model="defog/sqlcoder-7b-2")
+
+# Initialize the Groq client
+groq_client = Groq(api_key=os.environ.get("GROQ_API_KEY"))
+
+# Database connection details
+DB_CONFIG = {
+    'host': 'auth-db579.hstgr.io',
+    'user': 'u121769371_ki_aiml_test',
+    'password': os.environ.get("DB_PASSWORD"),
+    'database': 'u121769371_ki_aiml_test'
+}
+
+def generate_sql(text):
+    output = pipe(text, max_new_tokens=50)
+    return output[0]['generated_text']
+
+def execute_query(query):
+    try:
+        connection = mysql.connector.connect(**DB_CONFIG)
+        cursor = connection.cursor()
+        cursor.execute(query)
+        results = cursor.fetchall()
+        cursor.close()
+        connection.close()
+        return results
+    except mysql.connector.Error as err:
+        print(f"Error: {err}")
+        return None
+
+@app.route('/')
+def index():
+    return render_template('index.html')
+
+@app.route('/chatbot', methods=['POST'])
+def chatbot():
+    data = request.json
+    user_query = data.get('text')
+
+    if not user_query:
+        return jsonify({"error": "No query provided"}), 400
+
+    try:
+        # Step 1: Convert natural language to SQL
+        sql_query = generate_sql(user_query)
+
+        # Step 2: Execute SQL query
+        query_result = execute_query(sql_query)
+
+        if query_result is None:
+            return jsonify({"error": "Database query execution failed"}), 500
+
+        # Step 3: Generate natural language response using Groq
+        prompt = f"Original query: {user_query}\nSQL query: {sql_query}\nQuery result: {query_result}\nPlease provide a natural language summary of the query result."
+        
+        chat_completion = groq_client.chat.completions.create(
+            messages=[
+                {
+                    "role": "user",
+                    "content": prompt,
+                }
+            ],
+            model="llama3-8b-8192",
+        )
+
+        natural_language_response = chat_completion.choices[0].message.content
+
+        return jsonify({"response": natural_language_response})
+
+    except Exception as e:
+        return jsonify({"error": str(e)}), 500
+
+if name == 'main':
+    app.run(host='0.0.0.0', port=8000)

requirements.txt

@@ -0,0 +1,9 @@
+fastapi==0.74.*
+uvicorn[standard]==0.17.*
+requests==2.27.*
+transformers==4.*
+torch==1.11.*
+numpy<2.0.0
+mysql-connector-python==8.0.*
+groq==0.3.*
+pydantic==1.9.*

static/script.js

@@ -0,0 +1,51 @@
+document.addEventListener('DOMContentLoaded', (event) => {
+    const chatMessages = document.getElementById('chat-messages');
+    const userInput = document.getElementById('user-input');
+    const sendButton = document.getElementById('send-button');
+
+    function addMessage(message, isUser = false) {
+        const messageElement = document.createElement('p');
+        messageElement.textContent = message;
+        messageElement.style.backgroundColor = isUser ? '#e6f2ff' : '#f0f0f0';
+        messageElement.style.padding = '10px';
+        messageElement.style.borderRadius = '5px';
+        messageElement.style.marginBottom = '10px';
+        chatMessages.appendChild(messageElement);
+        chatMessages.scrollTop = chatMessages.scrollHeight;
+    }
+
+    function sendMessage() {
+        const message = userInput.value.trim();
+        if (message) {
+            addMessage(You: ${message}, true);
+            userInput.value = '';
+
+            fetch('/chatbot', {
+                method: 'POST',
+                headers: {
+                    'Content-Type': 'application/json',
+                },
+                body: JSON.stringify({ text: message }),
+            })
+            .then(response => response.json())
+            .then(data => {
+                if (data.error) {
+                    addMessage(Error: ${data.error});
+                } else {
+                    addMessage(Chatbot: ${data.response});
+                }
+            })
+            .catch((error) => {
+                console.error('Error:', error);
+                addMessage('Error: Unable to get response from the server.');
+            });
+        }
+    }
+
+    sendButton.addEventListener('click', sendMessage);
+    userInput.addEventListener('keypress', function (e) {
+        if (e.key === 'Enter') {
+            sendMessage();
+        }
+    });
+});

templates/index.html

@@ -0,0 +1,59 @@
+<!DOCTYPE html>
+<html lang="en">
+  <head>
+    <meta charset="UTF-8" />
+    <meta name="viewport" content="width=device-width, initial-scale=1.0" />
+    <title>SQL Chatbot</title>
+    <style>
+      body {
+        font-family: Arial, sans-serif;
+        max-width: 800px;
+        margin: 0 auto;
+        padding: 20px;
+        background-color: #f0f0f0;
+      }
+      #chat-container {
+        background-color: white;
+        border-radius: 10px;
+        padding: 20px;
+        box-shadow: 0 0 10px rgba(0, 0, 0, 0.1);
+      }
+      #chat-messages {
+        height: 400px;
+        overflow-y: auto;
+        border: 1px solid #ddd;
+        padding: 10px;
+        margin-bottom: 20px;
+      }
+      #user-input {
+        width: 100%;
+        padding: 10px;
+        border: 1px solid #ddd;
+        border-radius: 5px;
+        margin-bottom: 10px;
+      }
+      #send-button {
+        background-color: #4caf50;
+        color: white;
+        border: none;
+        padding: 10px 20px;
+        text-align: center;
+        text-decoration: none;
+        display: inline-block;
+        font-size: 16px;
+        margin: 4px 2px;
+        cursor: pointer;
+        border-radius: 5px;
+      }
+    </style>
+  </head>
+  <body>
+    <div id="chat-container">
+      <h1>SQL Chatbot</h1>
+      <div id="chat-messages"></div>
+      <input type="text" id="user-input" placeholder="Ask a question..." />
+      <button id="send-button">Send</button>
+    </div>
+    <script src="{{ url_for('static', filename='script.js') }}"></script>
+  </body>
+</html>