DrishtiSharma commited on
Commit
19a06bc
·
verified ·
1 Parent(s): 2275de5

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +26 -0
app.py ADDED
@@ -0,0 +1,26 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pandas as pd
3
+ import sqlite3
4
+ import os
5
+ import json
6
+ from pathlib import Path
7
+ from datetime import datetime, timezone
8
+ from crewai import Agent, Crew, Process, Task
9
+ from crewai_tools import tool
10
+ from langchain_groq import ChatGroq
11
+ from langchain.schema.output import LLMResult
12
+ from langchain_core.callbacks.base import BaseCallbackHandler
13
+ from langchain_community.tools.sql_database.tool import (
14
+ InfoSQLDatabaseTool,
15
+ ListSQLDatabaseTool,
16
+ QuerySQLCheckerTool,
17
+ QuerySQLDataBaseTool,
18
+ )
19
+ from langchain_community.utilities.sql_database import SQLDatabase
20
+ from datasets import load_dataset
21
+ import tempfile
22
+
23
+ # API Key
24
+ os.environ["GROQ_API_KEY"] = st.secrets.get("GROQ_API_KEY", "")
25
+
26
+ # Initialize LLM