Update app.py

app.py

@@ -84,6 +84,8 @@ def image_generation(query):
     return "output.jpg"
 
 # Function to handle different input types and choose the right tool
+from llama_index.core.chat_engine.types import AgentChatResponse
+
 def handle_input(user_prompt, image=None, audio=None, websearch=False):
     if audio:
         if isinstance(audio, str):
@@ -111,6 +113,7 @@ def handle_input(user_prompt, image=None, audio=None, websearch=False):
         messages = [{"role": "user", "content": [image, user_prompt]}]
         response = vqa_model.chat(image=None, msgs=messages, tokenizer=tokenizer)
     else:
+        # Modify this part to check if a tool is required or if a direct answer suffices
         response = agent.chat(user_prompt)
     
     # Extract the content from AgentChatResponse to return as a string