salami24 commited on
Commit
177ab8b
·
verified ·
1 Parent(s): 07abdfc

Upload 4 files

Browse files
src/app.py ADDED
@@ -0,0 +1,33 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ import pickle
3
+ from sklearn.metrics.pairwise import cosine_similarity
4
+
5
+ # Load model and dataset
6
+ with open("currentaffairs_chatbot_vectorizer.pkl", "rb") as f:
7
+ vectorizer = pickle.load(f)
8
+
9
+ with open("chatbot_dataset.pkl", "rb") as f:
10
+ df = pickle.load(f)
11
+
12
+ # Rebuild embeddings
13
+ X_vectors = vectorizer.transform(df["Question"])
14
+
15
+ # Chatbot function
16
+ def chatbot_response(user_input):
17
+ user_vec = vectorizer.transform([user_input])
18
+ similarities = cosine_similarity(user_vec, X_vectors)
19
+ idx = similarities.argmax()
20
+ return df.iloc[idx]["Answer"]
21
+
22
+ # Streamlit UI
23
+ st.set_page_config(page_title="🌍 UN Countries Chatbot", page_icon="🌍")
24
+
25
+ st.title("🌍 UN Countries Chatbot")
26
+ st.write("Ask me about any UN member state and I’ll tell you the basic info.")
27
+
28
+ # User input box
29
+ user_input = st.text_input("You:", "")
30
+
31
+ if user_input:
32
+ response = chatbot_response(user_input)
33
+ st.success(f"🤖 {response}")
src/chatbot_dataset.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a348b9c43cdcb34e3b5a52f1410c58cc26bf42b40191c475c23bb608e0feaa69
3
+ size 2547
src/currentaffairs_chatbot_vectorizer.pkl ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:79912a9d7846f325cc0db1eab791bb79a9ff7684948acc6aabf2ca9e26173d83
3
+ size 1177
src/requirements.txt ADDED
@@ -0,0 +1,4 @@
 
 
 
 
 
1
+ pip>=25.0
2
+ streamlit
3
+ scikit-learn
4
+ pandas