Update app.py

app.py

@@ -1,34 +1,43 @@
 import os
 import gradio as gr
-from langchain.chat_models import ChatOpenAI
-from langchain import LLMChain, PromptTemplate
+from langchain_openai import ChatOpenAI
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 
-OPENAI_API_KEY=os.getenv('OPENAI_API_KEY')
+# 🔑 API Key (must be set in Hugging Face Secrets, not here)
+# Hugging Face will automatically load it if you add OPENAI_API_KEY in "Settings > Variables and secrets"
+openai_api_key = os.getenv("OPENAI_API_KEY")
+
+# Define template
+template = """Meet Arun, your youthful and witty personal assistant! 
+At 21 years old, he is full of energy and always eager to help. 
+Arun's goal is to assist you with any questions or problems you might have. 
+His enthusiasm shines through in every response, making interactions enjoyable and engaging.
 
-template = """You are a helpful assistant to answer all user queries.
 {chat_history}
 User: {user_message}
 Chatbot:"""
 
-prompt = PromptTemplate(
-    input_variables=["chat_history", "user_message"], template=template
-)
-
+prompt = PromptTemplate(input_variables=["chat_history", "user_message"], template=template)
 memory = ConversationBufferMemory(memory_key="chat_history")
 
-llm_chain = LLMChain(
-    llm=ChatOpenAI(temperature='0.5', model_name="gpt-3.5-turbo"),
-    prompt=prompt,
-    verbose=True,
-    memory=memory,
-)
+# Chat model
+llm = ChatOpenAI(temperature=0.5, model="gpt-4o-mini", openai_api_key=openai_api_key)
+llm_chain = LLMChain(llm=llm, prompt=prompt, verbose=True, memory=memory)
 
-def get_text_response(user_message,history):
-    response = llm_chain.predict(user_message = user_message)
+# Chat function
+def get_text_response(user_message, history):
+    response = llm_chain.predict(user_message=user_message)
     return response
 
-demo = gr.ChatInterface(get_text_response)
+# Gradio UI
+demo = gr.ChatInterface(
+    fn=get_text_response,
+    examples=["How are you doing?", "What are your interests?", "Which places do you like to visit?"],
+    title="Arun - AI Assistant",
+    description="Your youthful, witty, personal chatbot"
+)
 
 if __name__ == "__main__":
-    demo.launch() #To create a public link, set `share=True` in `launch()`. To enable errors and logs, set `debug=True` in `launch()`.
+    demo.launch()