Added new abilities

Author: cxbxmxcx

logs/logs.txt

@@ -9,11 +9,11 @@
 from playground.actions_manager import ActionsManager
 from playground.assistants_api import api
 from playground.assistants_panel import assistants_panel
-from playground.assistants_utils import EventHandler
+from playground.assistants_utils import EventHandler, get_tools
 from playground.environment_manager import EnvironmentManager
 from playground.logging import Logger
 from playground.semantic_manager import SemanticManager
-from playground.constants import ASSISTANTS_WORKING_FOLDER
+from playground.global_values import GlobalValues
 
 thread = api.create_thread()  # create a new thread everytime this is run
 actions_manager = ActionsManager()
@@ -55,17 +55,35 @@ def print_like_dislike(x: gr.LikeData):
     print(x.index, x.value, x.liked)
 
 
-def ask_assistant(history, message):
+def ask_assistant(assistant_id, history, message):
+    assistant = api.retrieve_assistant(assistant_id)
+    if assistant is None:
+        history.append((None, "Assistant not found."))
+        return history, gr.MultimodalTextbox(value=None, interactive=False)
+
     if history is None:
         history = []
 
     attachments = []
     content = ""
     for file in message["files"]:
-        history.append(((file,), None))
-        # upload files to the thread
-        file = api.upload_file(file)
-        attachments += [{"file_id": file.id, "tools": [{"type": "code_interpreter"}]}]
+        tools, actions = get_tools(assistant.tools)
+        if "Code Interpreter" in tools:
+            # upload files to the thread
+            file = api.upload_file(file)
+            attachments += [
+                {"file_id": file.id, "tools": [{"type": "code_interpreter"}]}
+            ]
+        else:
+            with open(file, "r") as f:
+                file_content = f.read()
+            file = os.path.basename(file)
+            file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, file)
+            with open(file_path, "w") as file:
+                file.write(file_content)
+                attachments = None
+            history.append((f"file {file}, saved to working folder.", None))
+
     if message["text"] is not None:
         history.append((message["text"], None))
         content = message["text"]
@@ -97,7 +115,7 @@ def get_file_path(file):
     if os.path.isabs(file):
         return file
 
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, file)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, file)
     return file_path
 
 
@@ -248,7 +266,9 @@ def stream_worker(assistant_id, thread_id, event_handler):
                 )
 
                 chat_msg = chat_input.submit(
-                    ask_assistant, [chatbot, chat_input], [chatbot, chat_input]
+                    ask_assistant,
+                    [assistant_id, chatbot, chat_input],
+                    [chatbot, chat_input],
                 )
                 bot_msg = chat_msg.then(
                     run,

main.py

@@ -3,7 +3,7 @@
 import os
 
 from playground.actions_manager import agent_action
-from playground.constants import ASSISTANTS_WORKING_FOLDER
+from playground.global_values import GlobalValues
 
 
 @agent_action
@@ -18,7 +18,7 @@ def excel_to_csv(filename, skiprows):
     Returns:
     str: The CSV-formatted text string.
     """
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     # Read the Excel file and skip the specified rows
     df = pd.read_excel(file_path, skiprows=skiprows)
 
@@ -57,7 +57,7 @@ def csv_to_excel(csv_content, start_row, filename):
     df = pd.read_csv(buffer)
 
     # Save the DataFrame to an Excel file
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     df.to_excel(file_path, index=False, startrow=start_row)
 
     # Close the buffer

playground/assistant_actions/excel_actions.py

@@ -1,7 +1,7 @@
 import os
 
 from playground.actions_manager import agent_action
-from playground.constants import ASSISTANTS_WORKING_FOLDER
+from playground.global_values import GlobalValues
 
 
 @agent_action
@@ -12,7 +12,7 @@ def save_file(filename, content):
     :param filename: The name of the file including extension.
     :param content: The content to save in the file.
     """
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     with open(file_path, "w", encoding="utf-8") as file:
         file.write(content)
     return f"File '{filename}' saved successfully."
@@ -26,7 +26,7 @@ def load_file(filename):
     :param filename: The name of the file including extension.
     :return: The content of the file.
     """
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     if not os.path.exists(file_path):
         print(f"File '{filename}' does not exist.")
         return None
@@ -44,7 +44,7 @@ def delete_file(filename):
 
     :param filename: The name of the file including extension.
     """
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     if os.path.exists(file_path):
         os.remove(file_path)
         return f"File '{filename}' deleted successfully."
@@ -59,7 +59,7 @@ def create_folder(foldername):
 
     :param foldername: The name of the folder to create.
     """
-    folder_path = os.path.join(ASSISTANTS_WORKING_FOLDER, foldername)
+    folder_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, foldername)
     if not os.path.exists(folder_path):
         os.makedirs(folder_path)
         return f"Folder '{foldername}' created successfully."
@@ -74,10 +74,21 @@ def list_files():
 
     :return: A list of file names.
     """
-    files = os.listdir(ASSISTANTS_WORKING_FOLDER)
+    files = os.listdir(GlobalValues.ASSISTANTS_WORKING_FOLDER)
     return files
 
 
+@agent_action
+def set_working_folder(foldername):
+    """
+    Set the working folder for file operations.
+
+    :param foldername: The name of the folder to set as the working folder.
+    """
+    GlobalValues.set_value("ASSISTANTS_WORKING_FOLDER", foldername)
+    return f"Working folder set to '{foldername}'."
+
+
 # # Example usage:
 # if __name__ == "__main__":
 #     # Save a file

playground/assistant_actions/file_actions.py

@@ -5,7 +5,7 @@
 import base64
 
 from playground.actions_manager import agent_action
-from playground.constants import ASSISTANTS_WORKING_FOLDER
+from playground.global_values import GlobalValues
 
 load_dotenv()
 
@@ -38,7 +38,9 @@ def create_image(prompt, model="dall-e-3", size="1024x1024", quality="standard",
     image_response = requests.get(image_url)
     if image_response.status_code == 200:
         local_filename = f"{prompt.replace(' ', '_').replace(',','')[:50]}.png"
-        local_path = os.path.join(ASSISTANTS_WORKING_FOLDER, local_filename)
+        local_path = os.path.join(
+            GlobalValues.ASSISTANTS_WORKING_FOLDER, local_filename
+        )
         with open(local_path, "wb") as f:
             f.write(image_response.content)
         return local_filename
@@ -58,7 +60,7 @@ def encode_image(image_filename):
     Returns:
         str: The base64 encoded string of the image.
     """
-    local_path = os.path.join(ASSISTANTS_WORKING_FOLDER, image_filename)
+    local_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, image_filename)
     if not os.path.exists(local_path):
         return f"File not found: {image_filename}"
     with open(local_path, "rb") as image_file:

playground/assistant_actions/image_actions.py

@@ -1,7 +1,7 @@
 # This file contains the implementation of assistant actions related to PowerPoint presentations. It is still a work in progress.
 
 from playground.actions_manager import agent_action
-from playground.constants import ASSISTANTS_WORKING_FOLDER
+from playground.global_values import GlobalValues
 from pptx import Presentation
 from pptx.util import Inches
 import markdown
@@ -23,7 +23,7 @@ def create_new_powerpoint(filename):
         os.path.dirname(os.path.abspath(__file__)), "template.pptx"
     )
     prs = Presentation(template_path)
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     prs.save(file_path)
     return f"Created new PowerPoint presentation: {filename}"
 
@@ -76,7 +76,7 @@ def add_slide_to_powerpoint(filename, slide_content):
     Returns:
     None
     """
-    file_path = os.path.join(ASSISTANTS_WORKING_FOLDER, filename)
+    file_path = os.path.join(GlobalValues.ASSISTANTS_WORKING_FOLDER, filename)
     prs = Presentation(file_path)
     slide_type = slide_content.get("slide_type", "Title and Content")