Update app.py

app.py

@@ -1,4 +1,3 @@
-
 import os
 import subprocess
 import streamlit as st
@@ -312,10 +311,12 @@ class AIAgent:
         Handles user input and executes the corresponding action.
         """
         if input_str:
-         action, *args = input_str.split()
-     else:
-         # Handle empty input (e.g., display a message or do nothing)
-         print("Please provide input.")
+            action, *args = input_str.split()
+            if action in self.tools:
+                if args:
+                    self.task_history.append({
+                        "action": action,
+                        "input": " ".join(args),
                         "output": self.tools[action](" ".join(args))
                     })
                 else:
@@ -327,6 +328,9 @@ class AIAgent:
                 print(f"Action: {action}\nInput: {' '.join(args)}\nOutput: {self.tools[action](' '.join(args))}")
             else:
                 raise InvalidActionError("Invalid action. Please choose a valid action from the list of tools.")
+        else:
+            # Handle empty input (e.g., display a message or do nothing)
+            print("Please provide input.")
         except (InvalidActionError, InvalidInputError, CodeGenerationError, CodeRefinementError,
                 CodeTestingError, CodeIntegrationError, AppTestingError, WorkspaceExplorerError,
                 PromptManagementError, SearchError) as e: