Update app.py

app.py

@@ -80,10 +80,12 @@ def query_vectara(query: str, filter_str="", lambda_val=0.0) -> str:
     return re.sub(r'\[\d+(,\d+){0,5}\]', '', answer)
 
     
-
 # Streamlit UI
 st.title('Vectara Query Interface')
 
+# Dropdown for selecting case number
+case_number = st.selectbox("Select Case Number:", range(2001, 2100), format_func=lambda x: f"Case Number: {x}")
+
 # User input for query
 user_query = st.text_input("Enter your query:", "")
 
@@ -95,7 +97,7 @@ lambda_val = st.sidebar.slider("Lambda Value:", min_value=0.0, max_value=1.0, va
 if st.button('Search'):
     if user_query:
         with st.spinner('Querying Vectara...'):
-            output = query_vectara(user_query, filter_str, lambda_val)
+            output = query_vectara(user_query, case_number, filter_str, lambda_val)
             st.markdown("## Result")
             st.write(output)
     else:
@@ -103,9 +105,6 @@ if st.button('Search'):
 
 
 
-
-
-
 # Initialize the HHEM model +++++++++++++++++++++++++++++++++++++++++++++++
 model = CrossEncoder('vectara/hallucination_evaluation_model')