Update app.py

app.py

@@ -1,22 +1,30 @@
 import os
 import pandas as pd
 import logging
-from datasets import load_dataset
-from langchain.embeddings import HuggingFaceEmbeddings  # Updated import
-from langchain.vectorstores import Chroma  # Updated import
-from langchain.prompts import PromptTemplate  # Updated import
-from langchain.schema.output_parser import StrOutputParser  # Updated import
-from langchain.schema.runnable import RunnablePassthrough  # Updated import
-import gradio as gr
-from transformers import AutoTokenizer, AutoModelForCausalLM, pipeline, BitsAndBytesConfig
-from fastapi import FastAPI, Header, HTTPException
 import threading
+from fastapi import FastAPI, Header, HTTPException
 import uvicorn
+import gradio as gr
+
+# LangChain imports (updated to current structure)
+from langchain_community.embeddings import HuggingFaceEmbeddings
+from langchain_community.vectorstores import Chroma
+from langchain_core.prompts import PromptTemplate
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import RunnablePassthrough
+
+# Transformers and datasets
+from datasets import load_dataset
+from transformers import (
+    AutoTokenizer,
+    AutoModelForCausalLM,
+    pipeline,
+    BitsAndBytesConfig
+)
 
 # ====================== CONFIGURATION ======================
 API_KEY = "Samson"
 MODEL_NAME = "microsoft/phi-2" 
-# ===========================================================
 
 # Set up logging
 logging.basicConfig(level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')