Updated to access information from server

.env-sample

@@ -0,0 +1,3 @@
+HF_TOKEN=
+NOODLE_PUBLIC_API_URL=
+GAME_DEPLOYMENT_URL=

app.py

@@ -2,7 +2,11 @@ import gradio as gr
 import json
 import os
 from smol import ChatAgent
+from dotenv import load_dotenv, dotenv_values 
 
+load_dotenv() 
+
+GAME_DEPLOYMENT_URL= os.getenv("GAME_DEPLOYMENT_URL")
 GLOBAL_STATE_FILE = "global_state.json"
 if os.path.exists(GLOBAL_STATE_FILE):
     with open(GLOBAL_STATE_FILE, "r") as f:
@@ -40,7 +44,6 @@ def chat_function(user_prompt, history):
     # Clear the input and return the updated history
     return "", history
 
-
 with gr.Blocks() as demo:
     gr.Markdown("## Live Game & Chat Interface")
 
@@ -50,8 +53,8 @@ with gr.Blocks() as demo:
             gr.Markdown("### Game")
             # The iframe points to your static Game
             gr.HTML(
-                value="""
-                <iframe src="https://experiment-sandpack.vercel.app" style="width:100%; height:600px; border:none;"></iframe>
+                value=f"""
+                <iframe src="{GAME_DEPLOYMENT_URL}"style="width:100%; height:600px; border:none;"></iframe>
                 """
             )
         # Right Column: Chat interface (smaller column)

smol.py

@@ -3,7 +3,7 @@ import shutil
 
 from smolagents import CodeAgent, DuckDuckGoSearchTool, Tool, HfApiModel
 from dotenv import load_dotenv, dotenv_values 
-from tool import FindFilesTool, GitPushTool, FileReplaceTool, ProcessFlowIdentifierTool, GetImageDimensionsTool, FileModifyTool
+from tool import GetGameFileNamesTool, FindFilesTool, GitPushTool, FileReplaceTool, ProcessFlowIdentifierTool, GetImageDimensionsTool, FileModifyTool
 
 load_dotenv() 
 
@@ -28,6 +28,7 @@ step 4: run appropriate tool to accomplish task
 step 5: upload changes to github
 """
 
+get_game_file_names_tool = GetGameFileNamesTool()
 find_files_tool = FindFilesTool()
 file_replace_tool = FileReplaceTool()
 process_identifier_tool= ProcessFlowIdentifierTool()
@@ -37,33 +38,38 @@ file_modify_tool = FileModifyTool()
 def ChatAgent(userPrompt):
     #Identify purpose of prompt
     promptCleanerAgent = CodeAgent(tools=[], model=model)
+
     instructions = promptCleanerAgent.run(f"""
         determine the purpose of the following string "{userPrompt}" if it is one of the following: [asset_change, script_update, conversation].
         asset_change: The user wants to change an asset in the game.
         script_update: The user wants to update the game script.
         conversation: The user wants to have a conversation with the AI by asking general questions or greetings.
     """)
-    
+
     appDescription = """
-        This is a 2d platformer game where the player controls a ball that bounces off platforms falling down. This app
+        This is an ai python app that iframes a nextjs 2d platformer game. The goal of this app is to have the user
+        chat with an ai agent to modify the nextjs game. The user can ask the ai agent to modify the game assets, scripts, or have a conversation.
+        The game is a 2d platformer game where the player controls a ball that bounces off platforms which falling down. The nextjs app
         uses typescript and sandpack. The folder components/sandpack-examples.tsx file contains the game logic and scripts.
     """
+
     if(instructions == "conversation"):
         contextPrompt = f"""
             User prompt '{userPrompt}'
             1) Reply to the user as a friendly ai agent.
-            2) Do not use any tools to modify files.
+            2) Do not use any tools to modify files but you can use get_game_file_names_tool to check for game directories.
             3) End process after replying to the user.
         """
     else:
         contextPrompt = f'using process_identifier_tool look for the appropriate instructions for "{instructions}" and apply it to the user prompt after this'
 
-    agent = CodeAgent(tools=[find_files_tool, process_identifier_tool, image_generation_tool, file_modify_tool, get_image_dimensions_tool, file_replace_tool], model=model)
+    agent = CodeAgent(tools=[get_game_file_names_tool, find_files_tool, process_identifier_tool, image_generation_tool, file_modify_tool, get_image_dimensions_tool, file_replace_tool], model=model)
+
     if instructions == "conversation":
         response = agent.run(f"{contextPrompt}")
     else:
         response = agent.run(f"{appDescription} {contextPrompt} {userPrompt} ")
-    # agent = CodeAgent(tools=[DuckDuckGoSearchTool()], model=HfApiModel())
+
     # Step 1: Prompt reception
     print(f"Response made: {response}")
     return response

tool.py

@@ -1,6 +1,13 @@
 from smolagents import CodeAgent, DuckDuckGoSearchTool, Tool, HfApiModel
 import os
 from dotenv import load_dotenv
+import requests
+
+load_dotenv() 
+
+HF_TOKEN = os.getenv("HF_TOKEN")
+NOODLE_PUBLIC_API_URL = os.getenv("NOODLE_PUBLIC_API_URL")
+
 class FileReaderTool(Tool):
     name = "file_reader_tool"
     description = """
@@ -57,10 +64,6 @@ class FileModifyTool(Tool):
     output_type = "string"
 
     def forward(self,file_location, prompt) -> str:
-        load_dotenv() 
-
-        HF_TOKEN = os.getenv("HF_TOKEN")
-
         file_reader_tool = FileReaderTool()
         file_write_tool = FileWriteTool()
 
@@ -218,6 +221,25 @@ class GitPushTool(Tool):
         except subprocess.CalledProcessError as e:
             return print(f"An error occurred while performing Git operations: {e}")
 
+class GetGameFileNamesTool(Tool):
+    name="get_game_file_names_tool"
+    description="""
+        Use this tool to get the game directories of files related to the game. The output of this is an array of file paths. 
+        Return only the array of file paths returned by the api call.
+    """
+    inputs = {}
+    output_type = "string"
+    def forward(self) -> str:
+        get_files_api = NOODLE_PUBLIC_API_URL + "/api/files/"
+
+        response = requests.get(get_files_api)
+        if response.status_code == 200:
+            print(response.json()['content'])
+        else:
+            print("Failed to get file content")
+        return response.json()['content']
+        
+
 class FindFilesTool(Tool):
     name = "find_files"
     description = "Find files with a given extension in a directory and its subdirectories"