Update app.py

app.py

@@ -1,65 +1,51 @@
 import streamlit as st
-import sys
-import pkg_resources
+import json
 
-# Page configuration
-st.set_page_config(
-    page_title="Portfolio Chatbot Test",
-    page_icon="🤖",
-    layout="wide"
-)
+# Page config
+st.set_page_config(page_title="Chatbot Test", layout="wide")
 
-# Debug information
-st.write("Python version:", sys.version)
-st.write("Installed packages:")
-installed_packages = [f"{pkg.key} {pkg.version}" for pkg in pkg_resources.working_set]
-st.write(installed_packages)
+# Initialize chat history
+if "messages" not in st.session_state:
+    st.session_state.messages = []
 
-try:
-    import transformers
-    st.success("Successfully imported transformers!")
-    st.write("Transformers version:", transformers.__version__)
-except ImportError as e:
-    st.error(f"Error importing transformers: {str(e)}")
+# Load knowledge base
+def load_knowledge_base():
+    try:
+        with open("knowledge_base.json", "r", encoding="utf-8") as f:
+            return json.load(f)
+    except Exception as e:
+        st.error(f"Error loading knowledge base: {str(e)}")
+        return {}
 
-# Initialize session state
-if 'messages' not in st.session_state:
-    st.session_state.messages = []
+kb = load_knowledge_base()
 
-def main():
-    st.title("Portfolio Chatbot Testing Interface")
-    st.write("Basic testing version")
-    
-    # Create two columns for layout
-    col1, col2 = st.columns([2, 1])
+# Main UI
+st.title("Portfolio Chat Test")
+
+# Chat interface
+for message in st.session_state.messages:
+    with st.chat_message(message["role"]):
+        st.write(message["content"])
+
+# Chat input
+if prompt := st.chat_input("Ask me anything..."):
+    # Add user message
+    st.session_state.messages.append({"role": "user", "content": prompt})
     
-    with col1:
-        st.subheader("Chat Interface")
-        # Display chat messages from history
-        for message in st.session_state.messages:
-            with st.chat_message(message["role"]):
-                st.markdown(message["content"])
-        
-        # Accept user input
-        if prompt := st.chat_input("What would you like to know?"):
-            # Add user message to chat history
-            st.session_state.messages.append({"role": "user", "content": prompt})
-            
-            # Simple echo response for testing
-            response = f"You asked: {prompt}"
-            
-            # Display assistant response
-            with st.chat_message("assistant"):
-                st.markdown(response)
-            
-            # Add assistant response to chat history
-            st.session_state.messages.append({"role": "assistant", "content": response})
+    # Generate simple response from knowledge base
+    if "projects" in kb and "project" in prompt.lower():
+        response = "Here are my projects: " + ", ".join(kb["projects"].keys())
+    elif "skills" in prompt.lower():
+        response = "I have experience in Python, ML, and Data Analysis."
+    else:
+        response = "I understand you're asking about: " + prompt
     
-    with col2:
-        st.subheader("Testing Tools")
-        if st.button("Clear Chat"):
-            st.session_state.messages = []
-            st.experimental_rerun()
+    # Add assistant response
+    with st.chat_message("assistant"):
+        st.write(response)
+    st.session_state.messages.append({"role": "assistant", "content": response})
 
-if __name__ == "__main__":
-    main()
+# Clear chat button
+if st.sidebar.button("Clear Chat"):
+    st.session_state.messages = []
+    st.rerun()