Update app.py

app.py

@@ -128,20 +128,20 @@ async def handle_chat(user_input):
         context = extract_text_from_html(url)
         user_input = processed_query
     else:
-    # Custom prompt shims
-    if "translate" in user_input:
-        context=""
-        prompt=""
-
-    else: # Search task
-        search_start_time = time.time()
-        results = await brave_search(user_input)
-        search_end_time = time.time()
-        
-        documents = [desc.replace('<strong>', '').replace('</strong>', '') for _, desc, _ in results]
-    
-        context = "\n".join(documents)
+        # Custom prompt shims
+        if "translate" in user_input:
+            context=""
+            prompt=""
     
+        else: # Search task
+            search_start_time = time.time()
+            results = await brave_search(user_input)
+            search_end_time = time.time()
+            
+            documents = [desc.replace('<strong>', '').replace('</strong>', '') for _, desc, _ in results]
+        
+            context = "\n".join(documents)
+        
     generation_start_time = time.time()
     response = await query_teapot(prompt, context, user_input)