Update app.py

app.py

@@ -51,18 +51,32 @@ class AIAgent:
         self.description = description
         self.skills = skills
 
-    def generate_agent_response(self, prompt: str) -> str:
-        # For now, we'll use a simple response. In a real implementation, you'd use a more sophisticated model.
-        return f"Agent {self.name} responding to: {prompt}"
-
-    def create_agent_prompt(self) -> str:
-        skills_str = '\n'.join([f"* {skill}" for skill in self.skills])
-        agent_prompt = f"""
-As an elite expert developer, my name is {self.name}. I possess a comprehensive understanding of the following areas:
-{skills_str}
-I am confident that I can leverage my expertise to assist you in developing and deploying cutting-edge web applications. Please feel free to ask any questions or present any challenges you may encounter.
-"""
-        return agent_prompt
+    def generate_agent_response(self, prompt: str, project_name: str) -> str:
+        # Simple logic to demonstrate agent capabilities
+        if "create file" in prompt.lower():
+            file_name = re.search(r"create file (\w+\.?\w*)", prompt.lower())
+            if file_name:
+                return self.create_file(project_name, file_name.group(1), "# New file created")
+            else:
+                return "I'm sorry, I couldn't understand the file name. Please specify the file name clearly."
+        
+        elif "add code" in prompt.lower():
+            file_name = re.search(r"add code to (\w+\.?\w*)", prompt.lower())
+            if file_name:
+                code_to_add = prompt.split("add code to")[1].split(":", 1)[-1].strip()
+                return self.add_code_to_file(project_name, file_name.group(1), code_to_add)
+            else:
+                return "I'm sorry, I couldn't understand the file name. Please specify the file name clearly."
+        
+        elif "create project" in prompt.lower():
+            new_project_name = re.search(r"create project (\w+)", prompt.lower())
+            if new_project_name:
+                return self.create_project(new_project_name.group(1))
+            else:
+                return "I'm sorry, I couldn't understand the project name. Please specify the project name clearly."
+        
+        else:
+            return f"I understand you're asking about '{prompt}'. How can I assist you with this in the context of your project?"
 
     def create_project(self, project_name: str) -> str:
         project_path = os.path.join(PROJECT_ROOT, project_name)
@@ -73,7 +87,7 @@ I am confident that I can leverage my expertise to assist you in developing and
         else:
             return f"Project '{project_name}' already exists."
 
-    def add_file_to_project(self, project_name: str, file_name: str, content: str) -> str:
+    def create_file(self, project_name: str, file_name: str, content: str) -> str:
         project_path = os.path.join(PROJECT_ROOT, project_name)
         if not os.path.exists(project_path):
             return f"Project '{project_name}' does not exist."
@@ -82,24 +96,17 @@ I am confident that I can leverage my expertise to assist you in developing and
         with open(file_path, "w") as file:
             file.write(content)
         st.session_state.workspace_projects[project_name]['files'].append(file_name)
-        return f"File '{file_name}' added to project '{project_name}'."
+        return f"File '{file_name}' created in project '{project_name}'."
 
-    def modify_file_in_project(self, project_name: str, file_name: str, content: str) -> str:
+    def add_code_to_file(self, project_name: str, file_name: str, code: str) -> str:
         project_path = os.path.join(PROJECT_ROOT, project_name)
         file_path = os.path.join(project_path, file_name)
         if not os.path.exists(file_path):
             return f"File '{file_name}' does not exist in project '{project_name}'."
         
-        with open(file_path, "w") as file:
-            file.write(content)
-        return f"File '{file_name}' in project '{project_name}' has been modified."
-
-# Initialize some example agents
-example_agents = [
-    AIAgent("WebDev Wizard", "Full-stack web development expert", ["HTML", "CSS", "JavaScript", "React", "Node.js"]),
-    AIAgent("Data Scientist", "Machine learning and data analysis specialist", ["Python", "Pandas", "Scikit-learn", "TensorFlow"]),
-    AIAgent("DevOps Guru", "Infrastructure and deployment expert", ["Docker", "Kubernetes", "AWS", "CI/CD"])
-]
+        with open(file_path, "a") as file:
+            file.write("\n" + code)
+        return f"Code added to '{file_name}' in project '{project_name}'."
 
 def main():
     st.title("AI-Guided Workspace")
@@ -124,7 +131,7 @@ def main():
         st.subheader("Chat with AI Agent")
         user_input = st.text_input("Enter your message:")
         if st.button("Send"):
-            agent_response = example_agents[0].generate_agent_response(user_input)
+            agent_response = example_agents[0].generate_agent_response(user_input, st.session_state.current_project)
             st.session_state.chat_history.append((user_input, agent_response))
 
         # Display chat history
@@ -137,7 +144,7 @@ def main():
         file_name = st.text_input("File name:")
         code = st_ace(language="python", theme="monokai", key="code_editor")
         if st.button("Save File"):
-            result = example_agents[0].add_file_to_project(st.session_state.current_project, file_name, code)
+            result = example_agents[0].create_file(st.session_state.current_project, file_name, code)
             st.write(result)
 
         # Display project files
@@ -145,6 +152,10 @@ def main():
         if st.session_state.current_project in st.session_state.workspace_projects:
             for file in st.session_state.workspace_projects[st.session_state.current_project]['files']:
                 st.text(file)
+                if st.button(f"View {file}"):
+                    file_path = os.path.join(PROJECT_ROOT, st.session_state.current_project, file)
+                    with open(file_path, "r") as f:
+                        st.code(f.read())
 
     else:
         st.write("Please create or select a project to get started.")