Update app.py

app.py

@@ -43,11 +43,18 @@ I am confident that I can leverage my expertise to assist you in developing and
         return agent_prompt
 
     def autonomous_build(self, chat_history, workspace_projects):
-        """
-        Autonomous build logic based on chat history and workspace projects.
-        """
-        summary = "Chat History:\n" + "\n".join([f"User: {u}\nAgent: {a}" for u, a in chat_history])
-        summary += "\n\nWorkspace Projects:\n" + "\n".join([f"{p}: {details}" for p, details in workspace_projects.items()])
+    """
+    Autonomous build logic based on chat history and workspace projects.
+    """
+    summary = "Chat History:\n" + "\n".join([f"User: {u}\nAgent: {a}" for u, a in chat_history])
+    summary += "\n\nWorkspace Projects:\n" + '\n'.join([f"{p}: {', '.join(details.keys())}" for p, details in workspace_projects.items()])
+
+    # Use Hugging Face models for advanced logic
+    summarizer = pipeline("summarization")
+    summary_output = summarizer(summary, max_length=100, min_length=50, do_sample=False)[0]['summary_text']
+
+    sentiment_analyzer = pipeline("sentiment-analysis")
+    sentiment_output = sentiment_analyzer(summary)[0]
 
         # Use a Hugging Face model for more advanced logic (e.g., a summarization model)
         summarizer = pipeline("summarization")
@@ -58,14 +65,19 @@ I am confident that I can leverage my expertise to assist you in developing and
 # Function to save an agent's prompt to a file and commit to the Hugging Face repository
 def save_agent_to_file(agent):
     """Saves the agent's prompt to a file locally and then commits to the Hugging Face repository."""
-    if not os.path.exists(AGENT_DIRECTORY):
-        os.makedirs(AGENT_DIRECTORY)
-    file_path = os.path.join(AGENT_DIRECTORY, f"{agent.name}.txt")
-    config_path = os.path.join(AGENT_DIRECTORY, f"{agent.name}Config.txt")
-    with open(file_path, "w") as file:
+    agents_path = os.path.join(PROJECT_ROOT, AGENT_DIRECTORY)
+    if not os.path.exists(agents_path):
+        os.makedirs(agents_path)
+    
+    agent_file = os.path.join(agents_path, f"{agent.name}.txt")
+    config_file = os.path.join(agents_path, f"{agent.name}Config.txt")
+    
+    with open(agent_file, "w") as file:
         file.write(agent.create_agent_prompt())
-    with open(config_path, "w") as file:
+    
+    with open(config_file, "w") as file:
         file.write(f"Agent Name: {agent.name}\nDescription: {agent.description}")
+    
     st.session_state.available_agents.append(agent.name)
 
     commit_and_push_changes(f"Add agent {agent.name}")
@@ -73,9 +85,9 @@ def save_agent_to_file(agent):
 # Function to load an agent's prompt from a file
 def load_agent_prompt(agent_name):
     """Loads an agent prompt from a file."""
-    file_path = os.path.join(AGENT_DIRECTORY, f"{agent_name}.txt")
-    if os.path.exists(file_path):
-        with open(file_path, "r") as file:
+    agent_file = os.path.join(AGENT_DIRECTORY, f"{agent_name}.txt")
+    if os.path.exists(agent_file):
+        with open(agent_file, "r") as file:
             agent_prompt = file.read()
         return agent_prompt
     else:
@@ -113,11 +125,8 @@ def chat_interface_with_agent(input_text, agent_name):
         input_ids = input_ids[:, :max_input_length]
 
     # Generate chatbot response
-    outputs = model.generate(
-        input_ids, max_new_tokens=50, num_return_sequences=1, do_sample=True, pad_token_id=tokenizer.eos_token_id  # Set pad_token_id to eos_token_id
-    )
-    response = tokenizer.decode(outputs[0], skip_special_tokens=True)
-    return response
+    chatbot_response = generator(input_ids, max_length=150, min_length=30, do_sample=True)[0]['generated_text']
+    return chatbot_response
 
 # Workspace interface for creating projects
 def workspace_interface(project_name):
@@ -125,8 +134,7 @@ def workspace_interface(project_name):
     if not os.path.exists(PROJECT_ROOT):
         os.makedirs(PROJECT_ROOT)
     if not os.path.exists(project_path):
-        os.makedirs(project_path)
-        st.session_state.workspace_projects[project_name] = {"files": []}
+        os topscorer.session_state.workspace_projects[project_name] = {"files": []}
         st.session_state.current_state['workspace_chat']['project_name'] = project_name
         commit_and_push_changes(f"Create project {project_name}")
         return f"Project {project_name} created successfully."