File size: 3,622 Bytes
ddf9ea0 70cf054 2a68e5a ddf9ea0 3c6562d ddf9ea0 3c6562d ddf9ea0 3c6562d ddf9ea0 3c6562d ddf9ea0 2a68e5a ddf9ea0 2a68e5a ddf9ea0 2a68e5a ddf9ea0 2a68e5a ddf9ea0 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 |
import logging
import gradio as gr
from nlp import NLPToSQL
from db import Database
nlp = NLPToSQL()
db = Database()
logging.basicConfig(level=logging.INFO)
def chatbot(user_query):
try:
sql = nlp.query_to_sql(user_query)
logging.info(f"Generated SQL: {sql}")
result = db.execute_query(sql)
if 'error' in result:
return f"❌ Error: {result['error']}"
if not result['data']:
return "⚠️ No data found."
# Formatting SQL output
response = f"**Query:**\n```\n{sql}\n```\n\n"
response += "**Result:**\n"
response += "<table style='width:100%; border-collapse: collapse;'>"
response += "<tr style='background-color: #f2f2f2;'>"
for col in result['columns']:
response += f"<th style='border: 1px solid #ddd; padding: 8px;'>{col}</th>"
response += "</tr>"
for row in result['data']:
response += "<tr>"
for cell in row:
response += f"<td style='border: 1px solid #ddd; padding: 8px;'>{cell}</td>"
response += "</tr>"
response += "</table>"
return response
except Exception as e:
logging.error(f"Error: {str(e)}")
return f"❌ Error: {str(e)}"
demo = gr.Interface(
fn=chatbot,
inputs=gr.Textbox(lines=2, placeholder="Ask your database anything..."),
outputs="markdown",
title="SQL Chat Assistant",
description="Enter a natural language question, and get the corresponding SQL query & result."
)
if __name__ == "__main__":
demo.launch(server_name="0.0.0.0", server_port=7860, share=True)
# Flask-based application can also be used
# import logging
# from flask import Flask, request, render_template_string
# from .nlp import NLPToSQL
# from .db import Database
# app = Flask(__name__)
# nlp = NLPToSQL()
# db = Database()
# logging.basicConfig(level=logging.INFO)
# HTML_TEMPLATE = """
# <!DOCTYPE html>
# <html>
# <head><title> Chat Assistant </title></head>
# <body>
# <h1> Database Chat Assistant</h1>
# <form method="POST">
# <input type="text" name="query" placeholder= "Enter your query..." size="50">
# <button type="submit">Ask</button>
# </form>
# {% if response %}
# <h3> Response: </h3>
# <pre>{{ response }}</pre>
# {% endif %}
# {% if error %}
# <p style = "color:red;">{{ error }} </p>
# {% endif %}
# </body>
# </html>
# """
# @app.route("/", methods=["GET", "POST"])
# def index():
# if request.method == 'POST':
# user_query = request.form['query']
# try:
# sql = nlp.query_to_sql(user_query)
# logging.info(f"Generated SQL: {sql}")
# result = db.execute_query(sql)
# if 'error' in result:
# return render_template_string(HTML_TEMPLATE, error=result['error'])
# if not result['data']:
# return render_template_string(HTML_TEMPLATE, error="No data found")
# response = " | ".join(result['columns']) + "\n"
# response += "-"*50 + "\n"
# for row in result['data']:
# response += " | ".join(str(cell) for cell in row) + "\n"
# return render_template_string(HTML_TEMPLATE, response=response)
# except Exception as e:
# logging.error(f"Error: {str(e)}")
# return render_template_string(HTML_TEMPLATE, error=f"Error: {str(e)}")
# return render_template_string(HTML_TEMPLATE)
# if __name__ == "__main__":
# app.run(debug=True) |