Upload 3 files

deme.py

@@ -0,0 +1,47 @@
+import streamlit as st
+import torch
+from transformers import LLMForConditionalGeneration, LLMTokenizer
+import sqlite3
+
+# Load Hugging Face LLM2 model and tokenizer
+model_name = "microsoft/CodeGPT-small-py"
+tokenizer = LLMTokenizer.from_pretrained(model_name)
+model = LLMForConditionalGeneration.from_pretrained(model_name)
+
+# Function to generate SQL query
+def generate_sql_query(text):
+    input_ids = tokenizer.encode(text, return_tensors="pt")
+    outputs = model.generate(input_ids, max_length=100, do_sample=False)
+    generated_sql = tokenizer.decode(outputs[0], skip_special_tokens=True)
+    return generated_sql
+
+# Function to execute SQL query and retrieve results from the database
+def execute_query(sql_query):
+    conn = sqlite3.connect('C:/Users/Chovatiya.Parth/Desktop/SQL/superstore Creation.sql')
+    cursor = conn.cursor()
+    cursor.execute(sql_query)
+    results = cursor.fetchall()
+    conn.close()
+    return results
+
+# Streamlit UI
+def main():
+    st.title("SQL Chatbot")
+
+    user_query = st.text_input("Enter your query:")
+
+    if st.button("Submit"):
+        sql_query = generate_sql_query(user_query)
+        st.write("Generated SQL query:", sql_query)
+
+        try:
+            results = execute_query(sql_query)
+            st.write("Results from the database:")
+            for row in results:
+                st.write(row)
+        except Exception as e:
+            st.error("An error occurred while executing the SQL query.")
+            st.error(e)
+
+if __name__ == "__main__":
+    main()

requiremnts.txt

@@ -0,0 +1,4 @@
+streamlit
+torch==1.10.0
+transformers==4.12.2
+sqlite3

superstore Creation.sql

@@ -0,0 +1,56 @@
+create database super_store;
+use super_store;
+
+create table customer(Customer_ID varchar(30) primary key,	
+						Customer_Name varchar(50),
+                        Segment varchar(20),
+                        Country varchar(20),
+                        City varchar(20),
+                        State varchar(20),
+                        Postal_Code int(10),
+                        Region varchar(20));
+select * from customer;
+
+CREATE TABLE super_store.order(
+    Order_ID VARCHAR(30) PRIMARY KEY,
+    Order_Date DATE,
+    Ship_Date DATE,
+    Ship_Mode VARCHAR(50)
+);
+
+select * from super_store.order;
+
+
+CREATE TABLE Product (
+    Product_ID VARCHAR(30) PRIMARY KEY,
+    Category VARCHAR(50),
+    Sub_Category VARCHAR(50),
+    Product_Name VARCHAR(150)
+);
+select * from Product;
+
+
+-- Create Returned table with foreign key constraint
+CREATE TABLE Returned (
+    Order_ID VARCHAR(30),
+    Returned varchar(10),
+    PRIMARY KEY (Order_ID),
+    FOREIGN KEY (Order_ID) REFERENCES super_store.order(Order_ID)
+);
+select * from returned;
+
+
+CREATE TABLE Sales (
+  Order_ID varchar(30) NOT NULL,
+  Customer_ID varchar(30) NOT NULL,
+  Product_ID varchar(30) NOT NULL,
+  Sales DECIMAL(10,2) NOT NULL,
+  Quantity INT NOT NULL,
+  Discount DECIMAL(5,2),
+  Profit DECIMAL(10,2),
+  PRIMARY KEY (Order_ID, Customer_ID, Product_ID),
+  FOREIGN KEY (Order_ID) REFERENCES super_store.order(Order_ID),
+  FOREIGN KEY (Customer_ID) REFERENCES Customer(Customer_ID),
+  FOREIGN KEY (Product_ID) REFERENCES Product(Product_ID)
+);
+select * from sales;