Update app.py

app.py

@@ -25,12 +25,14 @@ query_instruction = "Query the most relevant text from the BMW website"
 embeddings = HuggingFaceInstructEmbeddings(model_name=model_name, embed_instruction=embed_instruction, query_instruction=query_instruction)
 chain = RetrievalQAWithSourcesChain.from_chain_type(llm, chain_type="stuff", retriever=db.as_retriever(), memory=memory)
 
-def chat(message, site,history):
+def chat(message, history):
     history = history or []
     response = ""
     try:
         response = chain.run(input=message)
         markdown = generate_markdown(response)
+    except Exception as e:
+        print(f"Erorr: {e}")
     history.append((message, markdown))
 
     return history, history
@@ -58,6 +60,6 @@ with gr.Blocks() as demo:
         btn = gr.Button("Run").style(full_width=False)
         state = gr.State()
         agent_state = gr.State()
-        btn.click(chat, [inp,site,state],[chatbot, state])
+        btn.click(chat, [inp, state],[chatbot, state])
 if __name__ == '__main__':
     demo.launch()