Update app.py

app.py

@@ -1,7 +1,6 @@
-from langchain_openai import ChatOpenAI
-from langchain_core.prompts import ChatPromptTemplate
-from langchain_core.output_parsers import StrOutputParser
-from langchain_community.llms import Ollama
+from langchain.llms import Ollama  # Corrected import
+from langchain.prompts import ChatPromptTemplate  # Corrected import
+from langchain.output_parsers import StrOutputParser  # Corrected import
 import streamlit as st
 import os
 from dotenv import load_dotenv
@@ -16,22 +15,26 @@ os.environ["LANGCHAIN_API_KEY"] = os.getenv("LANGCHAIN_API_KEY")
 # Prompt Template
 prompt = ChatPromptTemplate.from_messages(
     [
-        ("system", "You are a helpful assistant. Please respond to the user queries"),
+        ("system", "You are a helpful assistant. Please respond to the user queries."),
         ("user", "Question: {question}")
     ]
 )
 
-# Streamlit app setup
-st.title('Langchain Demo With LLAMA2 API')
-
-# User input
-input_text = st.text_input("Search the topic you want")
+# Initialize the LLM
+llm = Ollama(model="llama2")  # Ensure "llama2" is the correct model name
 
-# Ollama LLM (ensure the model is available, or access it through Hugging Face API)
-llm = Ollama(model="llama2")
+# Output Parser
 output_parser = StrOutputParser()
+
+# Chain setup
 chain = prompt | llm | output_parser
 
+# Streamlit app setup
+st.title('LangChain Demo with LLaMA2 API')
+
+# User input
+input_text = st.text_input("Enter the topic you want to search:")
+
 # Display result when user inputs text
 if input_text:
     try: