Update app.py

app.py

@@ -161,17 +161,17 @@ def get_prompt(docs, query):
   prompt = base_prompt+"; Context: "+context+"; Question: "+query+"; Answer:"
   return(prompt)
 
-def run_query(input_text, country, model_sel):
+def run_query(input_query, country, model_sel):
     # docs = get_docs(input_text, country)
-    docs = get_docs(query, country=country,vulnerability_cat=vulnerability_options)
+    docs = get_docs(input_query, country=country,vulnerability_cat=vulnerability_options)
     # st.write('Selected country: ', country) # Debugging country
     if model_sel == "chatGPT":
         # res = pipe.run(query=input_text, documents=docs)
-        res = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": get_prompt(docs, query=query)}])
+        res = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": get_prompt(docs, query=input_query)}])
         output = res["results"][0]
         references = get_refs(docs, output)
     else:
-        res = client.text_generation(get_prompt_llama2(docs, query=input_text), max_new_tokens=4000, temperature=0.01, model=model)
+        res = client.text_generation(get_prompt_llama2(docs, query=input_query), max_new_tokens=4000, temperature=0.01, model=model)
         output = res
         references = get_refs(docs, res)
     st.write('Response')
@@ -263,13 +263,13 @@ else:
 
 
 if selected_example == "-":
-    text = st.text_area('Enter your question in the text box below using natural language or select an example from above:')
+    input_query = st.text_area('Enter your question in the text box below using natural language or select an example from above:')
 else:
-    text = st.text_area('Enter your question in the text box below using natural language or select an example from above:', value=selected_example)
+    input_query = st.text_area('Enter your question in the text box below using natural language or select an example from above:', value=selected_example)
 
 
 if st.button('Submit'):
-    run_query(text, country=country, model_sel=model_sel)
+    run_query(input_query, country=country, model_sel=model_sel)