Update app.py

app.py

@@ -2,16 +2,12 @@ import streamlit as st
 import os
 
 # YOUR API KEY
-API_KEY = "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"
-os.environ["OPENAI_API_KEY"] = API_KEY
+os.environ["OPENAI_API_KEY"] = "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"
 
 from langchain_openai import ChatOpenAI
-import tiktoken
 
-@st.cache_resource
-def get_chatbot():
-    # Simple knowledge base - no complex splitting
-    knowledge = """
+# Knowledge base
+KNOWLEDGE = """
 SR University is located in Warangal, Telangana, India. 
 Computer Science program focuses on AI/ML, DSA, Java/Python, AWS/Azure, software engineering.
 
@@ -24,46 +20,49 @@ Internship tips:
 3. Apply startups: AngelList, Y Combinator
 4. Practice system design, behavioral interviews
 5. Target: Google, Microsoft, Hyderabad/Bangalore startups
-    """
-    
-    llm = ChatOpenAI(model="gpt-4o-mini", temperature=0)
-    
-    def rag_chat(query):
-        prompt = f"""
-        Use ONLY this context to answer: 
-        {knowledge}
-        
-        Question: {query}
-        
-        Answer concisely and accurately:
-        """
-        return llm.invoke(prompt).content
-    
-    return rag_chat
+"""
 
-# UI
-st.title("🧠 RAG Chatbot")
-st.info("💡 Answers about SR University, AI internships, projects")
+@st.cache_resource
+def init_chatbot():
+    return ChatOpenAI(model="gpt-4o-mini", temperature=0)
 
-chatbot = get_chatbot()
+st.title("🧠 RAG Chatbot")
+st.markdown("💡 Ask about SR University, AI internships, projects...")
 
+chatbot = init_chatbot()
 if "messages" not in st.session_state:
     st.session_state.messages = []
 
-# Chat history
+# Display chat messages
 for message in st.session_state.messages:
     with st.chat_message(message["role"]):
         st.markdown(message["content"])
 
-# New message
-if prompt := st.chat_input("Ask about university, internships..."):
-    st.session_state.messages.append({"role": "user", "content": prompt})
+# Chat input
+if query := st.chat_input("Type your question here..."):
+    # Add user message
+    st.session_state.messages.append({"role": "user", "content": query})
     with st.chat_message("user"):
-        st.markdown(prompt)
+        st.markdown(query)
     
+    # Generate response
     with st.chat_message("assistant"):
-        with st.spinner("Answering..."):
-            response = chatbot(prompt)
+        with st.spinner('Thinking...'):
+            prompt = f"""Use ONLY this context to answer: 
+
+{KNOWLEDGE}
+
+Question: {query}
+
+Answer:"""
+            
+            response = chatbot.invoke(prompt).content
             st.markdown(response)
     
+    # Add assistant response to chat history
     st.session_state.messages.append({"role": "assistant", "content": response})
+
+# Clear chat button
+if st.button("🗑️ Clear Chat"):
+    st.session_state.messages = []
+    st.rerun()