Update app.py

app.py

@@ -26,7 +26,7 @@ def get_post_headers() -> dict:
         "Content-Type": "application/json",
     }
 
-def query_vectara(query: str, case_number: int, filter_str="", lambda_val=0.0) -> str:
+def query_vectara(query: str, filter_str="", lambda_val=0.0) -> str:
     corpus_key = {
         "customerId": customer_id,
         "corpusId": corpus_id,
@@ -58,8 +58,8 @@ def query_vectara(query: str, case_number: int, filter_str="", lambda_val=0.0) -
     }
 
     response = requests.post(
+        "https://api.vectara.io/v1/query",
         headers=get_post_headers(),
-        url="https://api.vectara.io/v1/query",
         data=json.dumps(data),
         timeout=130,
     )
@@ -73,19 +73,12 @@ 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
+# Dropdown for selecting case number as the query
 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:", "")
@@ -93,15 +86,11 @@ lambda_val = st.sidebar.slider("Lambda Value:", min_value=0.0, max_value=1.0, va
 
 # 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.")
-
+    query_str = f"Case Number = {case_number}"
+    with st.spinner('Querying Vectara...'):
+        output = query_vectara(query_str, filter_str, lambda_val)
+        st.markdown("## Result")
+        st.write(output)