Update app.py

app.py

@@ -5,11 +5,12 @@ from langchain.chains import LLMChain
 from langchain.prompts import PromptTemplate
 from langchain.memory import ConversationBufferMemory
 
-# 🔑 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"
+# ✅ Load OpenAI key from Hugging Face secrets
 openai_api_key = os.getenv("OPENAI_API_KEY")
+if not openai_api_key:
+    raise ValueError("OPENAI_API_KEY not found. Please set it in Hugging Face → Settings → Variables and secrets.")
 
-# Define template
+# Prompt 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. 
@@ -19,25 +20,43 @@ His enthusiasm shines through in every response, making interactions enjoyable a
 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 (Conversation history)
 memory = ConversationBufferMemory(memory_key="chat_history")
 
-# 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)
+# LLM with API Key
+llm = ChatOpenAI(
+    temperature=0.5,
+    model="gpt-4o-mini",
+    api_key=openai_api_key
+)
+
+# LLM Chain
+llm_chain = LLMChain(
+    llm=llm,
+    prompt=prompt,
+    verbose=True,
+    memory=memory
+)
 
-# Chat function
+# Chat Function
 def get_text_response(user_message, history):
     response = llm_chain.predict(user_message=user_message)
     return response
 
-# Gradio UI
+# Gradio Chat App
 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"
+    examples=[
+        "How are you doing?",
+        "What are your interests?",
+        "Which places do you like to visit?"
+    ]
 )
 
 if __name__ == "__main__":
-    demo.launch()
+    demo.launch(share=True)