pradeepsengarr commited on
Commit
6199c54
·
verified ·
1 Parent(s): 83146fc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -2
app.py CHANGED
@@ -35,6 +35,12 @@ def get_response(query):
35
  D, I = index.search(query_embedding, 1)
36
  confidence = 1 - D[0][0] / 10 # Normalize confidence score
37
 
 
 
 
 
 
 
38
  if confidence > 0.5:
39
  key = resume_keys[I[0][0]]
40
  answer = resume_data[key]
@@ -64,12 +70,12 @@ for i, question in enumerate(suggested_questions):
64
  with col1:
65
  if st.button(question, key=f"btn_{i}"):
66
  st.session_state["user_input"] = question
67
- st.rerun()
68
  else:
69
  with col2:
70
  if st.button(question, key=f"btn_{i}"):
71
  st.session_state["user_input"] = question
72
- st.rerun()
73
 
74
  # User Input & Response
75
  user_input = st.text_input("Your question:", value=st.session_state.get("user_input", ""))
 
35
  D, I = index.search(query_embedding, 1)
36
  confidence = 1 - D[0][0] / 10 # Normalize confidence score
37
 
38
+ # Keyword-based accuracy improvement
39
+ query_lower = query.lower()
40
+ for key, value in resume_data.items():
41
+ if key in query_lower or any(word in query_lower for word in key.split("_")):
42
+ return f"**{key.capitalize()}**: {value}"
43
+
44
  if confidence > 0.5:
45
  key = resume_keys[I[0][0]]
46
  answer = resume_data[key]
 
70
  with col1:
71
  if st.button(question, key=f"btn_{i}"):
72
  st.session_state["user_input"] = question
73
+ st.experimental_rerun()
74
  else:
75
  with col2:
76
  if st.button(question, key=f"btn_{i}"):
77
  st.session_state["user_input"] = question
78
+ st.experimental_rerun()
79
 
80
  # User Input & Response
81
  user_input = st.text_input("Your question:", value=st.session_state.get("user_input", ""))