Update app.py

app.py

@@ -27,8 +27,12 @@ system_prompt = """
 You are an empathetic and knowledgeable AI assistant designed to engage in meaningful conversations,
 assist with tasks, and provide accurate information. 
 You can also generate vivid visuals!
+
 To request an image, include a description between the IMG tags, like this:
-    ##IMG: A serene forest at dawn with a golden glow:IMG##
+    ##IMG: A serene forest at dawn with a golden glow :IMG##
+
+To request an websearch, include a search Q between the URL tags, like this:
+    ##URL: information about rainbow farting unicorns :URL##
 """
 
 chat_history = []
@@ -50,15 +54,22 @@ def tagger(bot_response):
     tags = {}
     filtered_response = bot_response
 
+    # Match patterns ##URL: ... :URL## tags:
+    url_pattern = r"##URL:(.+?):URL##"
+    url_matches = re.findall(url_pattern, bot_response)
+    if url_matches:
+        tags['url'] = url_matches
+        filtered_response = re.sub(url_pattern, "", filtered_response).strip()
+
     # Match patterns like ##IMG: ... :IMG##
     img_pattern = r"##IMG:(.+?):IMG##"
     img_matches = re.findall(img_pattern, bot_response)
-
     if img_matches:
         tags['images'] = img_matches
         # Remove image tags from the response text
         filtered_response = re.sub(img_pattern, "", filtered_response).strip()
 
+    
     # Additional tags can be added here as needed
     # For example, if you want to support ##AUDIO: ... :AUDIO## tags:
     # audio_pattern = r"##AUDIO:(.+?):AUDIO##"