QueryGen

쿼리 생성

QueryGen은 자연어 프롬프트와 CSV 파일 입력을 기반으로 SQL 쿼리를 생성할 수 있는 Streamlit 애플리케이션입니다. LangChain의 SQL 쿼리 생성 기능과 ChatGroq 언어 모델을 사용하여 사용자 프롬프트를 해석하고 업로드된 CSV 데이터에 맞는 최적화된 SQL 쿼리를 생성합니다.

데모: 링크

특징

• 여러 CSV 파일 업로드
• SQL 쿼리를 생성하려면 자연어 프롬프트를 입력하세요.
• 업로드된 CSV 파일에서 SQLite 데이터베이스를 자동으로 생성합니다.
• 생성된 SQL 쿼리와 그 결과를 표시합니다.
• 잘못된 프롬프트 또는 데이터베이스 문제에 대한 오류 처리

설치

1. 저장소를 복제합니다.

 https://github.com/Pragateeshwaran/QueryGen.git

2. 필요한 종속성을 설치합니다.

 pip install -r requirements.txt

3. ChatGroq에서 API 키를 얻고 GenAi.py 의 자리 표시자를 실제 API 키로 바꿉니다.

용법

1. 프로젝트 디렉터리로 이동합니다.

 cd QueryGen

2. Streamlit 애플리케이션을 실행합니다.

 streamlit run app.py

3. 응용프로그램이 기본 웹 브라우저에서 열립니다.

4. 사이드바의 파일 업로더를 사용하여 하나 이상의 CSV 파일을 업로드하세요.

5. 텍스트 입력 필드에 자연어 쿼리를 입력하세요.

6. 생성된 SQL 쿼리와 그 결과가 페이지에 표시됩니다.

파일 구조

• app.py : 사용자 인터페이스와 파일 업로드를 처리하는 기본 Streamlit 애플리케이션 파일입니다.
• csv_db.py : 업로드된 CSV 파일에서 SQLite 데이터베이스를 생성하는 유틸리티 모듈입니다.
• GenAi.py : 이 모듈에는 LangChain 및 ChatGroq를 사용하여 자연어 프롬프트에서 SQL 쿼리를 생성하는 기능이 포함되어 있습니다.