Update app.py

app.py

@@ -72,6 +72,11 @@ def execute_command(command, cwd=None):
     stdout, stderr = process.communicate()
     return stdout + stderr
 
+def read_knowledge_base():
+    knowledge_file = os.path.join(os.path.dirname(__file__), 'knowledge.txt')
+    with open(knowledge_file, 'r') as file:
+        return file.read()
+
 @app.route("/")
 def index():
     return render_template("index.html")
@@ -87,9 +92,10 @@ def execute_code():
         if command.lower().startswith("ai:"):
             # Process command with Gemini AI
             ai_command = command[3:].strip()
-            response = chat.send_message(system_instruction + f"\nUser request: {ai_command}")
+            knowledge_base = read_knowledge_base()
+            response = chat.send_message(f"{system_instruction}\n\nKnowledge Base:\n{knowledge_base}\n\nUser request: {ai_command}")
             ai_result = response.text.strip()
-            
+
             # Execute the AI-suggested command or action
             if ai_result.startswith("CREATE_FILE:"):
                 filename = ai_result.split(":")[1]
@@ -103,22 +109,14 @@ def execute_code():
                 with open(filepath, 'w') as f:
                     f.write(content)
                 result = f"File {filename} created and edited successfully."
-            elif ai_result.startswith("!python -m pip install"):
-                result = execute_command(ai_result[1:])  # Remove the leading '!'
-            elif ai_result.startswith("git clone"):
-                result = execute_command(ai_result)
-            elif ai_result.startswith("cd "):
-                new_dir = os.path.join(current_dir, ai_result[3:])
-                if os.path.isdir(new_dir):
-                    current_dir = os.path.abspath(new_dir)
-                    result = f"Changed directory to: {current_dir}"
-                else:
-                    result = f"Error: Directory not found: {new_dir}"
-            elif ai_result.startswith("!python "):
+            elif ai_result.startswith("!"):
                 result = execute_command(ai_result[1:])  # Remove the leading '!'
+            elif ai_result.startswith("show files"):
+                files = os.listdir(current_dir)
+                result = "Files in current directory:\n" + "\n".join(files)
             else:
-                result = f"Unclear AI response: {ai_result}"
-            
+                result = execute_command(ai_result)
+
             return jsonify({"result": f"AI Executed: {ai_result}\n\nOutput:\n{result}"})
         elif command == "show files":
             files = os.listdir(current_dir)