Update app.py

app.py

@@ -1,105 +1,111 @@
-import os
-from langchain_openai import ChatOpenAI
-
-# Initialize ChatOpenAI without explicitly setting the API key
-# It will automatically use OPENAI_API_KEY from your environment variables
-llm = ChatOpenAI(model="gpt-3.5-turbo")
-import bs4
-from langchain import hub
-from langchain_chroma import Chroma
-from langchain_community.document_loaders import WebBaseLoader
-from langchain_core.output_parsers import StrOutputParser
-from langchain_core.runnables import RunnablePassthrough
-from langchain_openai import OpenAIEmbeddings
-from langchain_text_splitters import RecursiveCharacterTextSplitter
-from langchain.document_loaders import TextLoader
-import gradio as gr
-
-# Load and process documents
-loader = TextLoader("cleaned_yu_sgc_content.txt", encoding='utf-8')
-docs = loader.load()
-
-text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
-splits = text_splitter.split_documents(docs)
-vectorstore = Chroma.from_documents(documents=splits, embedding=OpenAIEmbeddings())
-retriever = vectorstore.as_retriever()
-
-# Helper functions
-def format_docs(docs):
-    """Format documents into a single string."""
-    return "\n\n".join(doc.page_content for doc in docs)
-
-def format_chat_history(history):
-    """Format chat history into a string."""
-    formatted_history = ""
-    for human, assistant in history:
-        formatted_history += f"\nHuman: {human}\nAssistant: {assistant}"
-    return formatted_history
-
-def generate_prompt(context, question, chat_history):
-    """Generate a custom prompt including chat history."""
-    return f"""You are a helpful assistant designed to answer questions about Yeshiva University's Career Center.
-    
-    Previous conversation history:
-    {chat_history}
-    
-    Use the following context to answer the question. If the context doesn't contain the relevant information,
-    you can provide general information about career center services.
-
-    Context: {context}
-
-    For questions about YU Career Center services (appointments, location, assistance):
-    - Use the information from the context if available
-    - Include this link when relevant: [Yeshiva University Career Center](https://www.yu.edu/sgc)
-    - Be specific, clear, and concise
-    - Maintain consistency with previous responses in the conversation
-    
-    If you cannot find the answer in the context, provide a general response based on the Career Center website.
-    If you cannot help at all, respond with: "Sorry, I'm not able to help with that, but feel free to ask me something else!"
-
-    Current Question: {question}
-    Response:"""
-
-
-def chatbot_response(message, history):
-    """Process user input and return chatbot response with history."""
-    try:
-        # Format the chat history
-        chat_history = format_chat_history(history)
-        
-        # Get relevant documents
-        relevant_docs = retriever.get_relevant_documents(message)
-        context = format_docs(relevant_docs)
-        
-        # Generate the prompt with history
-        prompt = generate_prompt(context, message, chat_history)
-        
-        # Get response from LLM
-        response = llm.invoke(prompt).content
-        
-        return response
-    except Exception as e:
-        return f"I apologize, but I encountered an error: {str(e)}. Please try again."
-
-
-
-# Create and launch Gradio interface
-iface = gr.ChatInterface(
-    chatbot_response,
-    title="YU Career Center Assistant",
-    description="""Get help with Yeshiva University Career Center services and information. 
-                Ask questions about appointments, services, locations, and more.""",
-    examples=[
-        "How can I schedule a career counseling appointment?",
-        "What services does the Career Center offer?",
-        "Where is the Career Center located?",
-        "What are the Career Center's hours of operation?",
-        "How can I access resume writing resources?"
-    ],
-    theme="default"
-)
-
-# Launch the interface
-print("Starting YU Career Center Chatbot...")
-print("Access the interface in your browser when the URL appears.")
+import os
+from langchain_openai import ChatOpenAI
+
+# Initialize ChatOpenAI without explicitly setting the API key
+# It will automatically use OPENAI_API_KEY from your environment variables
+
+from dotenv import load_dotenv
+import os
+from langchain_openai import ChatOpenAI
+
+load_dotenv()  # Load environment variables from .env file
+llm = ChatOpenAI(model="gpt-3.5-turbo")
+import bs4
+from langchain import hub
+from langchain_chroma import Chroma
+from langchain_community.document_loaders import WebBaseLoader
+from langchain_core.output_parsers import StrOutputParser
+from langchain_core.runnables import RunnablePassthrough
+from langchain_openai import OpenAIEmbeddings
+from langchain_text_splitters import RecursiveCharacterTextSplitter
+from langchain.document_loaders import TextLoader
+import gradio as gr
+
+# Load and process documents
+loader = TextLoader("cleaned_yu_sgc_content.txt", encoding='utf-8')
+docs = loader.load()
+
+text_splitter = RecursiveCharacterTextSplitter(chunk_size=1000, chunk_overlap=200)
+splits = text_splitter.split_documents(docs)
+vectorstore = Chroma.from_documents(documents=splits, embedding=OpenAIEmbeddings())
+retriever = vectorstore.as_retriever()
+
+# Helper functions
+def format_docs(docs):
+    """Format documents into a single string."""
+    return "\n\n".join(doc.page_content for doc in docs)
+
+def format_chat_history(history):
+    """Format chat history into a string."""
+    formatted_history = ""
+    for human, assistant in history:
+        formatted_history += f"\nHuman: {human}\nAssistant: {assistant}"
+    return formatted_history
+
+def generate_prompt(context, question, chat_history):
+    """Generate a custom prompt including chat history."""
+    return f"""You are a helpful assistant designed to answer questions about Yeshiva University's Career Center.
+    
+    Previous conversation history:
+    {chat_history}
+    
+    Use the following context to answer the question. If the context doesn't contain the relevant information,
+    you can provide general information about career center services.
+
+    Context: {context}
+
+    For questions about YU Career Center services (appointments, location, assistance):
+    - Use the information from the context if available
+    - Include this link when relevant: [Yeshiva University Career Center](https://www.yu.edu/sgc)
+    - Be specific, clear, and concise
+    - Maintain consistency with previous responses in the conversation
+    
+    If you cannot find the answer in the context, provide a general response based on the Career Center website.
+    If you cannot help at all, respond with: "Sorry, I'm not able to help with that, but feel free to ask me something else!"
+
+    Current Question: {question}
+    Response:"""
+
+
+def chatbot_response(message, history):
+    """Process user input and return chatbot response with history."""
+    try:
+        # Format the chat history
+        chat_history = format_chat_history(history)
+        
+        # Get relevant documents
+        relevant_docs = retriever.get_relevant_documents(message)
+        context = format_docs(relevant_docs)
+        
+        # Generate the prompt with history
+        prompt = generate_prompt(context, message, chat_history)
+        
+        # Get response from LLM
+        response = llm.invoke(prompt).content
+        
+        return response
+    except Exception as e:
+        return f"I apologize, but I encountered an error: {str(e)}. Please try again."
+
+
+
+# Create and launch Gradio interface
+iface = gr.ChatInterface(
+    chatbot_response,
+    title="YU Career Center Assistant",
+    description="""Get help with Yeshiva University Career Center services and information. 
+                Ask questions about appointments, services, locations, and more.""",
+    examples=[
+        "How can I schedule a career counseling appointment?",
+        "What services does the Career Center offer?",
+        "Where is the Career Center located?",
+        "What are the Career Center's hours of operation?",
+        "How can I access resume writing resources?"
+    ],
+    theme="default"
+)
+
+# Launch the interface
+print("Starting YU Career Center Chatbot...")
+print("Access the interface in your browser when the URL appears.")
 iface.launch(share=True)