Update app.py

app.py

@@ -8,13 +8,8 @@ import os
 import json
 from streamlit_ace import st_ace
 from agent import (
-    AppType,
-    createLlamaPrompt,
-    createSpace,
-    isPythonOrGradioAppPrompt,
-    isReactAppPrompt,
-    isStreamlitAppPrompt,
-    generateFiles,
+    AppType, createLlamaPrompt, createSpace, isPythonOrGradioAppPrompt,
+    isReactAppPrompt, isStreamlitAppPrompt, generateFiles,
 )
 
 import importlib
@@ -328,7 +323,7 @@ def generate_code(code_idea):
         str: The generated code.
     """
     # Generate code using the Hugging Face API
-    generated_code = code_generator(f"```python\n{code_idea}\n```", max_length=512)[0]['generated_text']
+    generated_code = code_generator(f"python\n{code_idea}\n", max_length=512)[0]['generated_text']
 
     st.session_state.current_state['toolbox']['generated_code'] = generated_code
     return generated_code
@@ -412,213 +407,27 @@ elif app_mode == "Tool Box":
     if st.button("Analyze Sentiment"):
         sentiment = sentiment_analysis(sentiment_text)
         st.write(f"Sentiment: {sentiment}")
-    # Function to translate code using the Hugging Face API
-    def translate_code(code, input_language, output_language):
-        """
-        Translates code from one programming language to another using a Hugging Face translation pipeline.
-        Args:
-            code (str): The code to be translated.
-            input_language (str): The source programming language.
-            output_language (str): The target programming language.
-            Returns:
-            str: The translated code.
-        """
-    # Define a mapping of programming languages to natural languages
-    language_map = {
-        "python": "English",
-        "javascript": "French",
-        "java": "German",
-        # Add more mappings as needed
-    }
-
-    # Function to translate code using the Hugging Face API
-def translate_code(code, input_language, output_language):
-    """
-    Translates code from one programming language to another using a Hugging Face translation pipeline.
-    Args:
-        code (str): The code to be translated.
-        input_language (str): The source programming language.
-        output_language (str): The target programming language.
-    Returns:
-        str: The translated code.
-    """
-    # Define a mapping of programming languages to natural languages
-    language_map = {
-        "python": "English",
-        "javascript": "French",
-        "java": "German",
-        # Add more mappings as needed
-    }
-
-    # Translate the code using the translation pipeline
-    translated_code = translator(code, src_lang=language_map[input_language], tgt_lang=language_map[output_language])[0]['translation_text']
-    
-    st.session_state.current_state['toolbox']['translated_code'] = translated_code
-    return translated_code
-
-# Function to generate code using the Hugging Face API
-def generate_code(prompt):
-    """
-    Generates code based on a given prompt using a Hugging Face code generation pipeline.
-    Args:
-        prompt (str): The prompt describing the code to be generated.
-    Returns:
-        str: The generated code.
-    """
-    generated_code = code_generator(prompt, max_length=100, num_return_sequences=1)[0]['generated_text']
-    st.session_state.current_state['toolbox']['generated_code'] = generated_code
-    return generated_code
-
-# Function to commit and push changes to the Hugging Face repository
-def commit_and_push_changes(commit_message):
-    """
-    Commits and pushes changes to the Hugging Face repository.
-    Args:
-        commit_message (str): The commit message.
-    """
-    try:
-        subprocess.run(["git", "add", "."], check=True)
-        subprocess.run(["git", "commit", "-m", commit_message], check=True)
-        subprocess.run(["git", "push", HUGGING_FACE_REPO_URL], check=True)
-    except subprocess.CalledProcessError as e:
-        st.error(f"Error committing and pushing changes: {e}")
-
-# Streamlit UI
-st.title("AI Agent Creator and Workspace Chat App")
-
-# Sidebar navigation
-st.sidebar.title("Navigation")
-app_mode = st.sidebar.selectbox("Choose the app mode", ["AI Agent Creator", "Tool Box", "Workspace Chat App"])
-
-# Add dark mode toggle
-if st.sidebar.checkbox("Dark Mode"):
-    st.markdown("""
-        <style>
-        .stApp {
-            background-color: #2b2b2b;
-            color: #ffffff;
-        }
-        </style>
-    """, unsafe_allow_html=True)
-
-if app_mode == "AI Agent Creator":
-    st.header("Create an AI Agent")
-    
-    col1, col2 = st.columns(2)
-    
-    with col1:
-        agent_name = st.text_input("Enter agent name:")
-        agent_description = st.text_area("Enter agent description:")
-        agent_skills = st.text_area("Enter agent skills (one per line):")
-        
-        if st.button("Create Agent"):
-            skills_list = agent_skills.split('\n')
-            new_agent = AIAgent(agent_name, agent_description, skills_list)
-            save_agent_to_file(new_agent)
-            st.success(f"Agent '{agent_name}' created successfully.")
-    
-    with col2:
-        st.subheader("Available Agents")
-        for agent in st.session_state.available_agents:
-            st.write(f"- {agent}")
-
-elif app_mode == "Tool Box":
-    st.header("AI-Powered Tools")
-    
-    tool = st.selectbox("Select a tool", ["Chat", "Terminal", "Code Editor", "Summarize Text", "Sentiment Analysis", "Translate Code", "Generate Code"])
-    
-    if tool == "Chat":
-        st.subheader("Chat with AI Assistant")
-        chat_input = st.text_area("Enter your message:")
-        agent_name = st.selectbox("Select an agent (optional):", [""] + st.session_state.available_agents)
-        
-        if st.button("Send"):
-            if agent_name:
-                chat_response = chat_interface_with_agent(chat_input, agent_name)
-            else:
-                chat_response = chat_interface(chat_input)
-            st.session_state.chat_history.append((chat_input, chat_response))
-            st.write(f"AI: {chat_response}")
-    
-    elif tool == "Terminal":
-        st.subheader("Terminal")
-        terminal_input = st.text_input("Enter a command:")
-        project_name = st.text_input("Enter project name (optional):")
-        
-        if st.button("Run Command"):
-            terminal_output = terminal_interface(terminal_input, project_name)
-            st.session_state.terminal_history.append((terminal_input, terminal_output))
-            st.code(terminal_output, language="bash")
-    
-    elif tool == "Code Editor":
-        st.subheader("Code Editor")
-        code = st_ace(
-            placeholder="Write your code here",
-            language="python",
-            theme="monokai",
-            keybinding="vscode",
-            font_size=14,
-            tab_size=4,
-            show_gutter=True,
-            show_print_margin=True,
-            wrap=False,
-            auto_update=False,
-        )
-        if st.button("Format & Lint"):
-            with st.spinner("Formatting and linting..."):
-                formatted_code, lint_message = code_editor_interface(code)
-            st.code(formatted_code, language="python")
-            st.info(lint_message)
-    
-    elif tool == "Summarize Text":
-        st.subheader("Summarize Text")
-        text_to_summarize = st.text_area("Enter text to summarize:")
-        if st.button("Summarize"):
-            summary = summarize_text(text_to_summarize)
-            st.write("Summary:", summary)
-    
-    elif tool == "Sentiment Analysis":
-        st.subheader("Sentiment Analysis")
-        text_to_analyze = st.text_area("Enter text for sentiment analysis:")
-        if st.button("Analyze Sentiment"):
-            sentiment = sentiment_analysis(text_to_analyze)
-            st.write("Sentiment:", sentiment)
-    
-    elif tool == "Translate Code":
-        st.subheader("Translate Code")
-        code_to_translate = st.text_area("Enter code to translate:")
-        input_language = st.selectbox("Select input language:", ["python", "javascript", "java"])
-        output_language = st.selectbox("Select output language:", ["python", "javascript", "java"])
-        if st.button("Translate"):
-            translated_code = translate_code(code_to_translate, input_language, output_language)
-            st.code(translated_code, language=output_language)
-    
-    elif tool == "Generate Code":
-        st.subheader("Generate Code")
-        code_prompt = st.text_area("Enter a prompt for code generation:")
-        if st.button("Generate Code"):
-            generated_code = generate_code(code_prompt)
-            st.code(generated_code, language="python")
 
+# Workspace Chat App
 elif app_mode == "Workspace Chat App":
     st.header("Workspace Chat App")
-    
+
     col1, col2 = st.columns(2)
-    
+
     with col1:
         st.subheader("Create a New Project")
         project_name = st.text_input("Enter project name:")
         if st.button("Create Project"):
             workspace_status = workspace_interface(project_name)
             st.success(workspace_status)
-        
+
         st.subheader("Add Code to Workspace")
         code_to_add = st.text_area("Enter code to add to workspace:")
         file_name = st.text_input("Enter file name (e.g. 'app.py'):")
         if st.button("Add Code"):
             add_code_status = add_code_to_workspace(project_name, code_to_add, file_name)
             st.success(add_code_status)
-    
+
     with col2:
         st.subheader("Chat with AI Assistant")
         chat_input = st.text_area("Enter your message for guidance:")
@@ -626,17 +435,17 @@ elif app_mode == "Workspace Chat App":
             chat_response = chat_interface(chat_input)
             st.session_state.chat_history.append((chat_input, chat_response))
             st.write(f"AI: {chat_response}")
-    
+
     st.subheader("Chat History")
     for user_input, response in st.session_state.chat_history:
         st.write(f"User: {user_input}")
         st.write(f"AI: {response}")
-    
+
     st.subheader("Terminal History")
     for command, output in st.session_state.terminal_history:
         st.write(f"Command: {command}")
         st.code(output, language="bash")
-    
+
     st.subheader("Workspace Projects")
     for project, details in st.session_state.workspace_projects.items():
         st.write(f"Project: {project}")
@@ -645,103 +454,5 @@ elif app_mode == "Workspace Chat App":
             st.write(f"- {file}")
 
 # Display the current state
-st.sidebar.subheader("Current State"),
-st.sidebar.json(st.session_state.current_state),
-st.header("Workspace Chat App")
-
-    # Project Workspace Creation
-    st.subheader("Create a New Project")
-    project_name = st.text_input("Enter project name:")
-    if st.button("Create Project"):
-        workspace_status = workspace_interface(project_name)
-        st.success(workspace_status)
-
-    # Add Code to Workspace
-    st.subheader("Add Code to Workspace")
-    code_to_add = st.text_area("Enter code to add to workspace:")
-    file_name = st.text_input("Enter file name (e.g. 'app.py'):")
-    if st.button("Add Code"):
-        add_code_status = add_code_to_workspace(project_name, code_to_add, file_name)
-        st.success(add_code_status)
-
-    # Terminal Interface with Project Context
-    st.subheader("Terminal (Workspace Context)")
-    terminal_input = st.text_input("Enter a command within the workspace:")
-    if st.button("Run Command"):
-        terminal_output = terminal_interface(terminal_input, project_name)
-        st.code(terminal_output, language="bash")
-
-    # Chat Interface for Guidance
-    st.subheader("Chat with CodeCraft for Guidance")
-    chat_input = st.text_area("Enter your message for guidance:")
-    if st.button("Get Guidance"):
-        chat_response = chat_interface(chat_input)
-        st.session_state.chat_history.append((chat_input, chat_response))
-        st.write(f"CodeCraft: {chat_response}")
-
-    # Display Chat History
-    st.subheader("Chat History")
-    for user_input, response in st.session_state.chat_history:
-        st.write(f"User: {user_input}")
-        st.write(f"CodeCraft: {response}")
-
-    # Display Terminal History
-    st.subheader("Terminal History")
-    for command, output in st.session_state.terminal_history:
-        st.write(f"Command: {command}")
-        st.code(output, language="bash")
-
-    # Display Projects and Files
-    st.subheader("Workspace Projects")
-    for project, details in st.session_state.workspace_projects.items():
-        st.write(f"Project: {project}")
-        for file in details['files']:
-            st.write(f"  - {file}")
-
-    # Chat with AI Agents
-    st.subheader("Chat with AI Agents")
-    selected_agent = st.selectbox("Select an AI agent", st.session_state.available_agents)
-    agent_chat_input = st.text_area("Enter your message for the agent:")
-    if st.button("Send to Agent"):
-        agent_chat_response = chat_interface_with_agent(agent_chat_input, selected_agent)
-        st.session_state.chat_history.append((agent_chat_input, agent_chat_response))
-        st.write(f"{selected_agent}: {agent_chat_response}")
-
-    # Automate Build Process
-    st.subheader("Automate Build Process")
-    if st.button("Automate"):
-        agent = AIAgent(selected_agent, "", [])  # Load the agent without skills for now
-        summary, next_step = agent.autonomous_build(st.session_state.chat_history, st.session_state.workspace_projects)
-        st.write("Autonomous Build Summary:")
-        st.write(summary)
-        st.write("Next Step:")
-        st.write(next_step)
-
-    # Advanced Code Editor (Optional)
-    st.subheader("Advanced Code Editor")
-    selected_file = st.selectbox("Select a file from the workspace", st.session_state.workspace_projects[project_name]['files'])
-    file_path = os.path.join(PROJECT_ROOT, project_name, selected_file)
-    if os.path.exists(file_path):
-        with open(file_path, "r") as file:
-            file_content = file.read()
-        code_editor = st_ace(
-            file_content,
-            language="python",
-            theme="monokai",
-            height=300,
-            key="ace_editor",
-        )
-        if st.button("Save Changes"):
-            with open(file_path, "w") as file:
-                file.write(code_editor)
-            st.success(f"Changes saved to {selected_file}")
-            commit_and_push_changes(f"Update {selected_file}")
-    else:
-        st.warning(f"File {selected_file} not found in the workspace.")
-
-# Display current state for debugging
 st.sidebar.subheader("Current State")
-st.sidebar.json(st.session_state.current_state)
-
-if __name__ == "__main__":
-    os.system("streamlit run app.py")
+st.sidebar.json(st.session_state.current_state)