Upload 4 files

ai.py

@@ -0,0 +1,40 @@
+from langchain_groq.chat_models import ChatGroq
+# from langchain_ollama.chat_models import ChatOllama
+from dotenv import load_dotenv
+load_dotenv()
+import sqlite3
+import os
+os.environ["GROQ_API_KEY"]=os.getenv("GROQ_API_KEY")
+
+llm=ChatGroq(model_name="llama-3.3-70b-versatile")
+
+def generate_sql_response(request):
+    prompt=f"""You are a SQL expert with decades of experience.using your experience and skills generate 
+    a sql query that generates the correct output for the below request.\n
+    <INST>You have access to two tables:[Employees (ID, Name, Department, Salary, Hire_Date), Departments (ID, Name, Manager)].
+    Return just the sql query, NO DESCRIPTION, NO DELIMITER</INST>\n\n
+    
+    <request>{request}</request>"""
+
+    response=llm.invoke(prompt)
+    return response.content
+def execute_query(query):
+    cur=sqlite3.connect("company.db").cursor()
+    res=cur.execute(query)
+    data=[]
+    for e in res:
+        data.append(e)
+    
+    return data
+
+def print_table(data):
+    prompt=f"""You are given a python list in which each element represents the output of an executed sql query.
+    your task is to format this data and return in tabular manner titled "Output".
+    Don't give the code for the same, only the output.
+    In case the list is empty simply return "No Matching Entities found.
+    <list>{data}</list>"""
+
+    response=llm.invoke(prompt)
+
+    return response.content
+

app.py

@@ -0,0 +1,16 @@
+import streamlit as st
+from db import main
+from ai import generate_sql_response,execute_query,print_table
+
+st.set_page_config(
+    page_title="Ajackus Assignment",
+)
+# if "db" not in st.session_state:
+#     main()
+user_input=st.text_area(label="Enter:")
+if st.button("Generate"):
+    query=generate_sql_response(user_input)
+    query_op=execute_query(query)
+    response=print_table(query_op)
+
+    st.write(response)

db.py

@@ -0,0 +1,52 @@
+import sqlite3
+def main():
+    # Connect to SQLite database (or create it if it doesn't exist)
+    conn = sqlite3.connect("company.db")
+    cursor = conn.cursor()
+
+    # Create Employees table
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS Employees (
+        ID INTEGER PRIMARY KEY,
+        Name TEXT NOT NULL,
+        Department TEXT NOT NULL,
+        Salary INTEGER NOT NULL,
+        Hire_Date TEXT NOT NULL
+    )
+    """)
+
+    # Create Departments table
+    cursor.execute("""
+    CREATE TABLE IF NOT EXISTS Departments (
+        ID INTEGER PRIMARY KEY,
+        Name TEXT NOT NULL,
+        Manager TEXT NOT NULL
+    )
+    """)
+
+    # Insert data into Employees table
+    cursor.executemany("""
+    INSERT INTO Employees (ID, Name, Department, Salary, Hire_Date)
+    VALUES (?, ?, ?, ?, ?)
+    """, [
+        (1, "Alice", "Sales", 50000, "2021-01-15"),
+        (2, "Bob", "Engineering", 70000, "2020-06-10"),
+        (3, "Charlie", "Marketing", 60000, "2022-03-20")
+    ])
+
+    # Insert data into Departments table
+    cursor.executemany("""
+    INSERT INTO Departments (ID, Name, Manager)
+    VALUES (?, ?, ?)
+    """, [
+        (1, "Sales", "Alice"),
+        (2, "Engineering", "Bob"),
+        (3, "Marketing", "Charlie")
+    ])
+
+    # Commit changes and close connection
+    conn.commit()
+    conn.close()
+
+if __name__=="__main__":
+    main()

requirements.txt

@@ -0,0 +1,7 @@
+streamlit
+langchain
+langchain_groq
+sqlalchemy
+ipykernel
+langchain_community
+langchain_ollama