Update app.py

app.py

@@ -9,6 +9,7 @@ from dotenv import load_dotenv
 # Load environment variables
 load_dotenv()
 
+# Set environment variables
 os.environ["LANGCHAIN_TRACING_V2"] = "true"
 os.environ["LANGCHAIN_API_KEY"] = os.getenv("LANGCHAIN_API_KEY")
 
@@ -21,15 +22,22 @@ prompt = ChatPromptTemplate.from_messages(
 )
 
 # Streamlit app
-st.title('Langchain Demo With LLAMA2 API')
+st.title('Langchain Demo With Ollama Llama2 API')
 input_text = st.text_input("Search the topic you want")
 
-# Ollama LLama2 LLM
-llm = Ollama(model="llama2")
+# Ollama LLama2 LLM with remote or local settings
+# Ensure the model points to a valid URL if not running locally
+llm = Ollama(
+    model="llama2",
+    server_url=os.getenv("OLLAMA_SERVER_URL", "http://localhost:11434")  # Add server URL to env
+)
 output_parser = StrOutputParser()
 chain = prompt | llm | output_parser
 
 # Display result when user inputs text
 if input_text:
-    response = chain.invoke({"question": input_text})
-    st.write(response)
+    try:
+        response = chain.invoke({"question": input_text})
+        st.write(response)
+    except Exception as e:
+        st.error(f"Error: {e}")