Update app.py

app.py

@@ -161,13 +161,13 @@ def get_prompt(docs, input_query):
   prompt = base_prompt+"; Context: "+context+"; Question: "+input_query+"; Answer:"
   return(prompt)
 
-def run_query(input_query, country, model_sel):
+def run_query(input_text, country, model_sel):
     # docs = get_docs(input_text, country)
-    docs = get_docs(input_query, country=country,vulnerability_cat=vulnerability_options)
+    docs = get_docs(input_text, 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, input_query)}])
+        res = openai.ChatCompletion.create(model="gpt-3.5-turbo", messages=[{"role": "user", "content": get_prompt(docs, input_text)}])
         output = res.choices[0].message.content
         references = get_refs(docs, output)
     # else:
@@ -263,13 +263,13 @@ else:
 
 
 if selected_example == "-":
-    input_query = st.text_area('Enter your question in the text box below using natural language or select an example from above:')
+    text = st.text_area('Enter your question in the text box below using natural language or select an example from above:')
 else:
-    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)
+    text = 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(input_query, country=country, model_sel=model_sel)
+    run_query(text, country=country, model_sel=model_sel)