Update app.py

app.py

@@ -29,31 +29,31 @@ messages = [
     AIMessage(content="I'm great thank you. How can I help you?")
 ]
 
-def handle_query(query: str, mode: str):
+def handle_message(message: str, mode: str):
     
     # Check if query is empty
-    if not query.strip():
+    if not message.strip():
         return "Enter a valid message."
     if mode == "Chat":
-        return chat_mode(query)
+        return chat_mode(message)
     elif mode == "Web-Search":
-        return web_search(query)
+        return web_search(message)
     else:
         return "Select a valid mode."
 
-def chat_mode(query: str):
+def chat_mode(message: str):
     global messages
-    prompt = HumanMessage(content=query)
+    prompt = HumanMessage(content=message)
     messages.append(prompt)
     response = chat_model.invoke(messages)
     messages.append(response.content)
     if len(messages) >= 6:
         messages = messages[-6:]
-    return f"You: {query}\n\nIT-Assistant: {response.content}"
+    return f"You: {message}\n\nIT-Assistant: {response.content}"
 
-def web_search(query: str):
+def web_search(message: str):
     global messages
-    similar_docs = db.similarity_search(query, k=3)
+    similar_docs = db.similarity_search(message, k=3)
     if similar_docs:
         source_knowledge = "\n".join([x.page_content for x in similar_docs])
     else:
@@ -61,7 +61,7 @@ def web_search(query: str):
     augmented_prompt = f"""
     If the answer to the next query is not contained in the Web Search say 'No Answer Available' and then give advice.
 
-    Query: {query}
+    Query: {message}
 
     Web Search:
     {source_knowledge}
@@ -72,10 +72,10 @@ def web_search(query: str):
     messages.append(response.content)
     if len(messages) >= 6:
         messages = messages[-6:]
-    return f"You: {query}\n\nIT-Assistant: {response.content}"
+    return f"You: {message}\n\nIT-Assistant: {response.content}"
 
 demo = gr.Interface(
-    fn=handle_query,
+    fn=handle_message,
     inputs=["text", gr.Radio(["Chat", "Web-Search"], label="Mode", info="Choose a mode and enter your message, then click submit to interact.")],
     outputs="text",
     title="IT Assistant")