Update app.py

app.py

@@ -99,7 +99,7 @@ def chat_interface_with_agent(input_text, agent_name):
 
     # Combine the agent prompt with user input
     combined_input = f"{agent_prompt}\n\nUser: {input_text}\nAgent:"
-    
+
     # Truncate input text to avoid exceeding the model's maximum length
     max_input_length = 900
     input_ids = tokenizer.encode(combined_input, return_tensors="pt")
@@ -178,13 +178,22 @@ def generate_code(code_idea):
             {"role": "user", "content": f"Generate a Python code snippet for the following idea:\n\n{code_idea}"}
         ]
     st.session_state.current_state['toolbox']['generated_code'] = generated_code
-    
+
     return generated_code
-    
+
 def translate_code(code, input_language, output_language):
     # Define a dictionary to map programming languages to their corresponding file extensions
     language_extensions = {
-        
+        "Python": "py",
+        "JavaScript": "js",
+        "Java": "java",
+        "C++": "cpp",
+        "C#": "cs",
+        "Ruby": "rb",
+        "Go": "go",
+        "PHP": "php",
+        "Swift": "swift",
+        "TypeScript": "ts",
     }
 
     # Add code to handle edge cases such as invalid input and unsupported programming languages
@@ -307,11 +316,11 @@ elif app_mode == "Tool Box":
     # 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'):")
+    input_language = st.text_input("Enter input language (e.g. 'Python'):")
+    output_language = st.text_input("Enter output 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())
+        translated_code = translate_code(code_to_translate, input_language, output_language)
+        st.code(translated_code, language=output_language.lower())
 
     # Code Generation
     st.subheader("Code Generation")
@@ -384,28 +393,6 @@ elif app_mode == "Workspace Chat App":
     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)
-
-# Display current state for debugging
-st.sidebar.subheader("Current State")
-st.sidebar.json(st.session_state.current_state)
+        st.write("Files:")
+        for file in details["files"]:
+            st.write(f"- {file}")