Update app.py

app.py

@@ -1,18 +1,24 @@
-import json
+
+import streamlit as st
+
+# Access Hugging Face API key from secrets
+hf_token = st.secrets["huggingface"]["hf_token"]
+if not hf_token:
+    st.error("Hugging Face API key not found. Please make sure it is set in the secrets.")
+Example Code Using Secrets
+Here is the updated code snippet using the secrets from secrets.toml:
+
+PYTHON
 import streamlit as st
 import os
 import subprocess
 from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 from huggingface_hub import HfApi
 
-
-try:
-    huggingface_json = st.secrets["hf_token"]
-    huggingface_data = json.loads("huggingface_json")
-    huggingface_token = huggingface_data["huggingface_json"]
-except Exception as e:
-    st.error(f"Unable to load secrets: {str(e)}")
-    raise
+# Access Hugging Face API key from secrets
+hf_token = st.secrets["huggingface"]["hf_token"]
+if not hf_token:
+    st.error("Hugging Face API key not found. Please make sure it is set in the secrets.")
 
 PROJECT_ROOT = "projects"
 AGENT_DIRECTORY = "agents"
@@ -36,13 +42,13 @@ class AIAgent:
         self.name = name
         self.description = description
         self.skills = skills
-        self._hf_api = HfApi(token=huggingface_token)
 
     def create_agent_prompt(self):
         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
@@ -54,12 +60,8 @@ I am confident that I can leverage my expertise to assist you in developing and
         return summary, next_step
 
     def deploy_built_space_to_hf(self):
-        # Implement logic to deploy the built space to Hugging Face Spaces
         pass
 
-    def has_valid_hf_token(self):
-        return self._hf_api.whoami() is not None
-
 def process_input(input_text):
     chatbot = pipeline("text-generation", model="microsoft/DialoGPT-medium", tokenizer="microsoft/DialoGPT-medium")
     response = chatbot(input_text, max_length=50, num_return_sequences=1)[0]['generated_text']
@@ -99,86 +101,74 @@ def display_workspace_projects(workspace_projects):
     return "\n".join([f"{p}: {details}" for p, details in workspace_projects.items()])
 
 if __name__ == "__main__":
-    st.set_page_config(layout="wide", page_title="AI-Powered Development Platform")
-
-    # Sidebar
-    st.sidebar.title("AI Development Platform")
-    st.sidebar.header("Tool Box")
-    st.sidebar.subheader("Workspace Management")
-    project_name_input = st.sidebar.text_input("Project Name:")
-    create_project_button = st.sidebar.button("Create Project")
-    if create_project_button:
-        st.sidebar.write(workspace_interface(project_name_input))
-
-    st.sidebar.subheader("Code Generation")
-    selected_model = st.sidebar.selectbox("Select Code Model", AVAILABLE_CODE_GENERATIVE_MODELS)
-    code_input = st.sidebar.text_area("Enter Code Prompt:")
-    generate_code_button = st.sidebar.button("Generate Code")
-    if generate_code_button:
-        if selected_model:
-            tokenizer = AutoTokenizer.from_pretrained(selected_model)
-            model = AutoModelForCausalLM.from_pretrained(selected_model)
-            code_generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
-            generated_code = code_generator(code_input, max_length=500, num_return_sequences=1)[0]['generated_text']
-            st.sidebar.code(generated_code)
-        else:
-            st.sidebar.error("Please select a code model.")
-
-    st.sidebar.subheader("Terminal")
-    command = st.sidebar.text_input("Enter a command:")
-    if command:
-        output, error = run_code(command)
-        if error:
-            st.sidebar.error(f"Error executing command: {error}")
-        else:
-            st.sidebar.code(output)
-
-    # Main Content
-    st.title("AI-Powered Development Platform")
-    st.header("Workspace")
-    st.subheader("Chat with an AI Agent")
-    chat_input = st.text_input("Enter your message:")
-    if chat_input:
-        st.session_state.chat_history.append((chat_input, process_input(chat_input)))
-
-    st.markdown("## Chat History ##")
-    st.markdown(display_chat_history(st.session_state.chat_history))
-
-    st.subheader("Available Agents")
-    for agent_name in st.session_state.available_agents:
-        st.write(f"**{agent_name}**")
-
-    st.subheader("Project Management")
-    st.markdown(display_workspace_projects(st.session_state.workspace_projects))
-
-    # AI Guide
-    if ai_guide_level == "Full Assistance":
-        st.markdown("## AI Guide: Full Assistance ##")
-        st.write("**Recommended Action:**")
-        st.write("Create a new project and then generate some code.")
-    elif ai_guide_level == "Partial Assistance":
-        st.markdown("## AI Guide: Partial Assistance ##")
-        st.write("**Tips:**")
-        st.write("Use the chat interface to ask questions about your project.")
-        st.write("Use the code generation tool to generate code snippets.")
-    else:
-        st.markdown("## AI Guide: No Assistance ##")
-        st.write("You are on your own!")
-
-    # Autonomous Build
-    if st.button("Autonomous Build"):
-        project_name = project_name_input
-        selected_model = selected_model
-        agent = AIAgent("Code Architect", "I am an expert in code generation and deployment.", ["Code Generation", "Deployment"])
-        summary, next_step = agent.autonomous_build(st.session_state.chat_history, st.session_state.workspace_projects, project_name, selected_model, huggingface_token)
-        st.write("Autonomous Build Summary:")
-        st.write(summary)
-        st.write("Next Step:")
-        st.write(next_step)
-        if agent._hf_api and agent.has_valid_hf_token():
-            repository = agent.deploy_built_space_to_hf()
-            st.markdown("## Congratulations! Successfully deployed Space �� ##")
-            st.markdown("[Check out your new Space here](hf.co/" + repository.name + ")")
+    st.sidebar.title("Navigation")
+    app_mode = st.sidebar.selectbox("Choose the app mode", ["Home", "Terminal", "Explorer", "Code Editor", "Build & Deploy"])
+
+    if app_mode == "Home":
+        st.title("Welcome to AI-Guided Development")
+        st.write("This application helps you build and deploy applications with the assistance of an AI Guide.")
+        st.write("Toggle the AI Guide from the sidebar to choose the level of assistance you need.")
+
+    elif app_mode == "Terminal":
+        st.header("Terminal")
+        terminal_input = st.text_input("Enter a command:")
+        if st.button("Run"):
+            output = run_code(terminal_input)
+            st.session_state.terminal_history.append((terminal_input, output))
+            st.code(output, language="bash")
+        if ai_guide_level != "No Assistance":
+            st.write("Run commands here to add packages to your project. For example: `pip install <package-name>`.")
+            if terminal_input and "install" in terminal_input:
+                package_name = terminal_input.split("install")[-1].strip()
+                st.write(f"Package `{package_name}` will be added to your project.")
+
+    elif app_mode == "Explorer":
+        st.header("Explorer")
+        uploaded_file = st.file_uploader("Upload a file", type=["py"])
+        if uploaded_file:
+            file_details = {"FileName": uploaded_file.name, "FileType": uploaded_file.type}
+            st.write(file_details)
+            save_path = os.path.join(PROJECT_ROOT, uploaded_file.name)
+            with open(save_path, "wb") as f:
+                f.write(uploaded_file.getbuffer())
+            st.success(f"File {uploaded_file.name} saved successfully!")
+
+        st.write("Drag and drop files into the 'app' folder.")
+        for project, details in st.session_state.workspace_projects.items():
+            st.write(f"Project: {project}")
+            for file in details['files']:
+                st.write(f"  - {file}")
+                if st.button(f"Move {file} to app folder"):
+                    # Logic to move file to 'app' folder
+                    pass
+        if ai_guide_level != "No Assistance":
+            st.write("You can upload files and move them into the 'app' folder for building your application.")
+
+    elif app_mode == "Code Editor":
+        st.header("Code Editor")
+        code_editor = st.text_area("Write your code:", height=300)
+        if st.button("Save Code"):
+            # Logic to save code
+            pass
+        if ai_guide_level != "No Assistance":
+            st.write("The function `foo()` requires the `bar` package. Add it to `requirements.txt`.")
+
+    elif app_mode == "Build & Deploy":
+        st.header("Build & Deploy")
+        project_name_input = st.text_input("Enter Project Name for Automation:")
+        if st.button("Automate"):
+            selected_agent = st.selectbox("Select an AI agent", st.session_state.available_agents)
+            selected_model = st.selectbox("Select a code-generative model", AVAILABLE_CODE_GENERATIVE_MODELS)
+            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, project_name_input, selected_model, hf_token)
+            st.write("Autonomous Build Summary:")
+            st.write(summary)
+            st.write("Next Step:")
+            st.write(next_step)
+            if agent._hf_api and agent.has_valid_hf_token():
+                repository = agent.deploy_built_space_to_hf()
+                st.markdown("## Congratulations! Successfully deployed Space 🚀 ##")
+                st.markdown("[Check out your new Space here](hf.co/" + repository.name + ")")
 
     # CSS for styling
     st.markdown("""
@@ -191,14 +181,17 @@ if __name__ == "__main__":
         margin: 0;
         padding: 0;
     }
+
     h1, h2, h3, h4, h5, h6 {
         color: #333;
     }
+
     .container {
         width: 90%;
         margin: 0 auto;
         padding: 20px;
     }
+
     /* Navigation Sidebar */
     .sidebar {
         background-color: #2c3e50;
@@ -211,21 +204,25 @@ if __name__ == "__main__":
         width: 250px;
         overflow-y: auto;
     }
+
     .sidebar a {
         color: #ecf0f1;
         text-decoration: none;
         display: block;
         padding: 10px 0;
     }
+
     .sidebar a:hover {
         background-color: #34495e;
         border-radius: 5px;
     }
+
     /* Main Content */
     .main-content {
         margin-left: 270px;
         padding: 20px;
     }
+
     /* Buttons */
     button {
         background-color: #3498db;
@@ -236,9 +233,11 @@ if __name__ == "__main__":
         cursor: pointer;
         font-size: 16px;
     }
+
     button:hover {
         background-color: #2980b9;
     }
+
     /* Text Areas and Inputs */
     textarea, input[type="text"] {
         width: 100%;
@@ -248,10 +247,12 @@ if __name__ == "__main__":
         border-radius: 5px;
         box-sizing: border-box;
     }
+
     textarea:focus, input[type="text"]:focus {
         border-color: #3498db;
         outline: none;
     }
+
     /* Terminal Output */
     .code-output {
         background-color: #1e1e1e;
@@ -260,6 +261,7 @@ if __name__ == "__main__":
         border-radius: 5px;
         font-family: 'Courier New', Courier, monospace;
     }
+
     /* Chat History */
     .chat-history {
         background-color: #ecf0f1;
@@ -268,17 +270,21 @@ if __name__ == "__main__":
         max-height: 300px;
         overflow-y: auto;
     }
+
     .chat-message {
         margin-bottom: 10px;
     }
+
     .chat-message.user {
         text-align: right;
         color: #3498db;
     }
+
     .chat-message.agent {
         text-align: left;
         color: #e74c3c;
     }
+
     /* Project Management */
     .project-list {
         background-color: #ecf0f1;
@@ -287,13 +293,16 @@ if __name__ == "__main__":
         max-height: 300px;
         overflow-y: auto;
     }
+
     .project-item {
         margin-bottom: 10px;
     }
+
     .project-item a {
         color: #3498db;
         text-decoration: none;
     }
+
     .project-item a:hover {
         text-decoration: underline;
     }