Update modules/search_web.py

modules/search_web.py

@@ -26,8 +26,8 @@ class search:
                 urls.append(result['embed_url'])
         output_dict = {}
         output_dict["llm_output"] = "Done. Video's have been searched for and displayed."
-        output_dict["real_output"] = {"display": video_embeds, "name": num, "metadata": urls}
-        output_dict["type"] = "html_video"
+        output_dict["display"] = {"files": urls, "metadata": None}
+        output_dict["type"] = "video"
         return output_dict
     def search_text(self, query, max_results=5, top=5):
         results = self.ddgs.text(query, max_results=max_results, region="us-en")
@@ -46,10 +46,10 @@ class search:
             final_text += "\n"
             final_text += txt
             final_text += "\n"
-        final_text += f"\nThe text above is stored as the variable {num}"
+        #final_text += f"\nThe text above is stored as the variable {num}"
         output_dict = {}
         output_dict["llm_output"] = str(final_text)
-        output_dict["real_output"] = {"display": None, "name": num, "metadata": None}
+        output_dict["display"] = None
         output_dict["type"] = "text"
         return output_dict
     def search_images(self, keywords, max_results, region="us-en", safesearch="off", size=None, color=None, type_image=None, layout=None, license_image=None):
@@ -74,8 +74,8 @@ class search:
             urls.append(img['image'])
         output_dict = {}
         output_dict["llm_output"] = f"Done. Images's have been searched for and displayed. Stored as variable {num}"
-        output_dict["real_output"] = {"display": images, "name": num, "metadata": urls}
-        output_dict["type"] = "html_image"
+        output_dict["display"] = {"html": urls}
+        output_dict["type"] = "image"
         return output_dict
     def search_news(self, keywords, region="us-en", safesearch="off", timelimit="m", max_results=5):
         results = self.ddgs.news(keywords=keywords, region=region, safesearch=safesearch, timelimit=timelimit, max_results=max_results)
@@ -93,6 +93,6 @@ class search:
             news_list += news
         news_list += f"\nAll the text above is stored as a variable {num}."
         output_dict["llm_output"] = str(news_list)
-        output_dict["real_output"] = {"display": images, "name": num, "metadata": None}
+        output_dict["display"] = {"display": images}
         output_dict["type"] = "news"
         return output_dict