llm bugfix

app.py

@@ -1,7 +1,7 @@
 import streamlit as st
 from streamlit import session_state as session
 from src.laion_clap.inference import AudioEncoder
-from src.utils.spotify import SpotifyHandler, SpotifyAuthentication
+# from src.utils.spotify import SpotifyHandler, SpotifyAuthentication
 import pandas as pd
 from dotenv import load_dotenv
 from langchain.llms import CTransformers, Ollama

requirements.txt

@@ -87,4 +87,6 @@ wandb==0.16.0
 webdataset==0.2.77
 wget==3.2
 youtube-search-python==1.6.6
-python-dotenv
+python-dotenv
+langchain
+pymongo

src/laion_clap/inference.py

@@ -2,8 +2,6 @@ from tqdm import tqdm
 import librosa
 import torch
 from src import laion_clap
-import json
-import jmespath
 from ..config.configs import ProjectPaths
 
 

src/llm/chain.py

@@ -1,10 +1,9 @@
-from langchain.prompts import ChatPromptTemplate, PromptTemplate
+from langchain.prompts import ChatPromptTemplate
 from langchain.schema.runnable import RunnableLambda
 from operator import itemgetter
 from langchain.output_parsers import PydanticOutputParser
 from .output_parser import SongDescriptions
 from langchain.llms.base import LLM
-import json
 
 
 class LLMChain: