Update app.py

app.py

@@ -10,8 +10,38 @@ ds_model = ChatHuggingFace(llm = deep_seek_model, repo_id = 'deepseek-ai/DeepSee
 
 st.title("ChatGenius Hub: Master Every Data Skills")
 
-message = [SystemMessage(content = "Answer like a data scientist"),
-          HumanMessage(content = "Tell me about Support Vector")]
-result =  ds_model.invole(message)
-print(result.content)
+# message = [SystemMessage(content = "Answer like a data scientist"),
+#           HumanMessage(content = "Tell me about Support Vector")]
+# result =  ds_model.invoke(message)
+# print(result.content)
 
+
+if "messages" not in st.session_state:
+    st.session_state.messages = [
+        SystemMessage(content="Answer like a data scientist")
+    ]
+
+def generate_response(user_input):
+    # Append user message
+    st.session_state.messages.append(HumanMessage(content=user_input))
+    # Invoke the model
+    response = ds_model.invoke(st.session_state.messages)
+    # Append AI response
+    st.session_state.messages.append(AIMessage(content=response))
+    return response
+
+# User input
+user_input = st.text_input("Ask a question about Data Science:")
+
+if user_input:
+    with st.spinner("Getting answer..."):
+        answer = generate_response(user_input)
+    st.markdown(f"**Answer:** {answer}")
+
+# Display chat history
+if st.session_state.messages:
+    for msg in st.session_state.messages[1:]:  # skip initial SystemMessage
+        if isinstance(msg, HumanMessage):
+            st.markdown(f"**You:** {msg.content}")
+        elif isinstance(msg, AIMessage):
+            st.markdown(f"**Bot:** {msg.content}")