Update app.py

app.py

@@ -124,13 +124,18 @@ async def handle_chat(user_input):
     
     # If there's a URL, fetch the context
     if url:
+        search_start_time = time.time()
         context = await extract_text_from_html(url)
         user_input = processed_query
+        search_end_time = time.time()
     else:
         # Custom prompt shims
         if "translate" in user_input:
+            search_start_time = time.time()
             context=""
             prompt=""
+            search_end_time = time.time()
+            
     
         else: # Search task
             search_start_time = time.time()