Update app.py

app.py

@@ -73,7 +73,39 @@ def query_vectara(query: str, case_number: int, filter_str="", lambda_val=0.0) -
     answer = result["responseSet"][0]["summary"][0]["text"]
     return re.sub(r'\[\d+(,\d+){0,5}\]', '', answer)
 
-    
+
+# Title
+st.title('Vectara Query Interface')
+
+# Combine case number selection with query input
+case_number = st.selectbox("Select Case Number:", range(2001, 2099), format_func=lambda x: f"Case Number = {x}")
+
+# User input for query
+user_query = st.text_input("Enter your query:", "")
+
+# Display case number with query input
+combined_query = f"{user_query} for {case_number}" if user_query else ""
+
+# Advanced options
+st.sidebar.header("Advanced Options")
+filter_str = st.sidebar.text_input("Filter String:", "")
+lambda_val = st.sidebar.slider("Lambda Value:", min_value=0.0, max_value=1.0, value=0.025)
+
+# Search button
+if st.button('Search'):
+    if combined_query:
+        with st.spinner('Querying Vectara...'):
+            # Assuming `query_vectara` is a function that takes the combined query, filter string, and lambda value
+            output = query_vectara(combined_query, filter_str, lambda_val)
+            st.markdown("## Result")
+            st.write(output)
+    else:
+        st.error("Please enter a query to search.")
+
+
+
+
+""" 
 # Streamlit UI
 st.title('Vectara Query Interface')
 
@@ -98,7 +130,7 @@ if st.button('Search'):
     else:
         st.error("Please enter a query to search.")
 
-
+"""
 
 # Initialize the HHEM model +++++++++++++++++++++++++++++++++++++++++++++++
 model = CrossEncoder('vectara/hallucination_evaluation_model')