Update app.py

app.py

@@ -2,19 +2,18 @@
 import streamlit as st
 import os
 from dotenv import load_dotenv
-from langchain_huggingface import HuggingFaceEndpoint
+from langchain.chat_models import ChatOpenAI
 
 # Load HuggingFace API token
 load_dotenv()
-HF_TOKEN = os.getenv("HF_TOKEN")
+OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 
 # Initialize the HuggingFace LLM
-llm = HuggingFaceEndpoint(
-    repo_id="mistralai/Mistral-7B-Instruct-v0.3",
-    huggingfacehub_api_token=HF_TOKEN.strip(),
+llm = ChatOpenAI(
+    openai_api_key=OPENAI_API_KEY,
+    model_name="gpt-4o-mini",
     temperature=0.7,
-    task="text-generation",
-    max_new_tokens=200
+    max_tokens=50
 )
 
 # Streamlit UI setup