Update src/app.py

src/app.py

@@ -1,19 +1,34 @@
 import streamlit as st
-from ollama import Client
+from langchain_community.llms import Ollama
+from langchain.chains import LLMChain
+from langchain.prompts import PromptTemplate
 
-st.set_page_config(page_title="Ollama Chat", layout="centered")
-st.title("🧠 Chat with Ollama via Python Client")
+st.set_page_config(page_title="🧪 LangChain + Ollama Notebook")
 
-client = Client(host="http://localhost:11434")
+st.title("📓 LangChain + Ollama Jupyter-like App")
 
-prompt = st.text_area("Enter your prompt:", height=150)
-submit = st.button("Generate")
+# Setup LangChain LLM with Ollama backend
+llm = Ollama(model="llama3", base_url="http://localhost:11434")
 
-if submit and prompt.strip():
-    with st.spinner("Thinking..."):
-        try:
-            response = client.generate(model="llama3", prompt=prompt)
-            st.markdown("### Response")
-            st.write(response['response'])
-        except Exception as e:
-            st.error(f"Error: {e}")
+# Notebook-style prompt block
+code_cell = st.text_area("🧠 Enter your notebook-style cell (text or code):", height=200)
+run_button = st.button("▶️ Run")
+
+# Prompt template with notebook context
+template = """
+You are a helpful AI assistant embedded in a Jupyter-style notebook.
+Interpret the input below and respond helpfully.
+
+Input:
+{cell_input}
+
+Response:"""
+
+prompt = PromptTemplate(template=template, input_variables=["cell_input"])
+chain = LLMChain(llm=llm, prompt=prompt)
+
+if run_button and code_cell.strip():
+    with st.spinner("Running cell..."):
+        result = chain.run(cell_input=code_cell)
+        st.markdown("### 📬 Response")
+        st.write(result)