Update app.py

app.py

@@ -17,6 +17,12 @@ from agent import (
     generateFiles,
 )
 
+import importlib
+
+# Dynamically import the Code symbol from the agent module
+code_module = importlib.import_module('agent.Code')
+Code = getattr(code_module, 'Code')
+
 # Set Hugging Face repository URL and project root path
 HUGGING_FACE_REPO_URL = "https://huggingface.co/spaces/acecalisto3/Mistri"
 PROJECT_ROOT = "projects"
@@ -406,40 +412,241 @@ 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
+    }
 
-    # Text Translation Tool (Code Translation)
-    st.subheader("Translate Code")
-    code_to_translate = st.text_area("Enter code to translate:")
-    source_language = st.text_input("Enter source language (e.g. 'Python'):")
-    target_language = st.text_input("Enter target language (e.g. 'JavaScript'):")
-    if st.button("Translate Code"):
-        translated_code = translate_code(code_to_translate, source_language, target_language)
-        st.code(translated_code, language=target_language.lower())
-
-    # Code Generation
-    st.subheader("Code Generation")
-    code_idea = st.text_input("Enter your code idea:")
-    if st.button("Generate Code"):
-        generated_code = generate_code(code_idea)
-        st.code(generated_code, language="python")
-
-    # Display Preset Commands
-    st.subheader("Preset Commands")
-    preset_commands = {
-        "Create a new project": "create_project('project_name')",
-        "Add code to workspace": "add_code_to_workspace('project_name', 'code', 'file_name')",
-        "Run terminal command": "terminal_interface('command', 'project_name')",
-        "Generate code": "generate_code('code_idea')",
-        "Summarize text": "summarize_text('text')",
-        "Analyze sentiment": "sentiment_analysis('text')",
-        "Translate code": "translate_code('code', 'source_language', 'target_language')",
+    # 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
     }
-    for command_name, command in preset_commands.items():
-        st.write(f"{command_name}: `{command}`")
 
-# Workspace Chat App
+    # 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")
+
 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:")
+        if st.button("Get Guidance"):
+            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}")
+        st.write("Files:")
+        for file in details["files"]:
+            st.write(f"- {file}")
+
+# Display the current state
+st.sidebar.subheader("Current State")
+st.sidebar.json(st.session_state.current_state)t.header("Workspace Chat App")
 
     # Project Workspace Creation
     st.subheader("Create a New Project")