Update app.py

app.py

@@ -5,6 +5,14 @@ from transformers import pipeline, AutoModelForCausalLM, AutoTokenizer
 import black
 from pylint import lint
 from io import StringIO
+import re
+from typing import Dict, Any
+from langchain.embeddings import HuggingFaceEmbeddings
+from langchain.vectorstores import FAISS
+from langchain.llms import OpenAI
+from langchain.chains import ConversationChain
+from concurrent.futures import ThreadPoolExecutor
+import openai
 
 HUGGING_FACE_REPO_URL = "https://huggingface.co/spaces/acecalisto3/Mistri"
 PROJECT_ROOT = "projects"
@@ -24,6 +32,8 @@ if 'current_state' not in st.session_state:
         'toolbox': {},
         'workspace_chat': {}
     }
+if 'current_agent' not in st.session_state:
+    st.session_state.current_agent = None
 
 class AIAgent:
     def __init__(self, name, description, skills):
@@ -34,21 +44,25 @@ class AIAgent:
     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.
+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
 
     def autonomous_build(self, chat_history, workspace_projects):
-        """
-        Autonomous build logic that continues based on the state of chat history and workspace projects.
-        """
+        """Autonomous build logic that continues based on the state of 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()])
 
-        next_step = "Based on the current state, the next logical step is to implement the main application logic."
-
+        # Use an LLM to generate a summary of the current state and suggest the next step
+        model_name = "gpt-3.5-turbo"  # You can choose a different LLM if you prefer
+        response = openai.ChatCompletion.create(
+            model=model_name,
+            messages=[
+                {"role": "system", "content": "You are a helpful AI assistant that can help developers with building projects. "},
+                {"role": "user", "content": f"Based on the following summary, what is the next logical step in the development process? \n\n{summary}"}
+            ]
+        )
+        next_step = response.choices[0].message['content']
         return summary, next_step
 
 def save_agent_to_file(agent):
@@ -93,35 +107,85 @@ def chat_interface_with_agent(input_text, agent_name):
         model = AutoModelForCausalLM.from_pretrained(model_name)
         tokenizer = AutoTokenizer.from_pretrained(model_name)
         generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
-    except EnvironmentError as e:
-        return f"Error loading model: {e}"
+        response = generator(
+            f"{agent_prompt}\n\nUser: {input_text}\nAgent:",
+            max_length=1024,
+            num_return_sequences=1,
+            do_sample=True,
+            top_k=50,
+            temperature=0.7,
+        )[0]["generated_text"]
+        return response
+    except Exception as e:
+        return f"Error: {e}"
+
+# Chat interface for CodeCraft
+def chat_interface(input_text):
+    """Handles interactions with the CodeCraft AI agent."""
+    # Define a pre-defined prompt for CodeCraft
+    codecraft_prompt = """
+    You are CodeCraft, a helpful and knowledgeable AI assistant specializing in software development. 
+    You are designed to provide guidance, code snippets, and solutions to developers. 
+    Please answer user questions in a comprehensive and informative manner.
+    """
+
+    # Load the GPT-2 model
+    model_name = "gpt2"  # You can use a more advanced model like GPT-3 or GPT-4 if you have access
+    model = AutoModelForCausalLM.from_pretrained(model_name)
+    tokenizer = AutoTokenizer.from_pretrained(model_name)
+    generator = pipeline("text-generation", model=model, tokenizer=tokenizer)
 
     # 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")
-    if input_ids.shape[1] > max_input_length:
-        input_ids = input_ids[:, :max_input_length]
+    combined_input = f"{codecraft_prompt}\n\nUser: {input_text}\nCodeCraft:"
 
     # 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)
+    response = generator(
+        combined_input,
+        max_length=1024,
+        num_return_sequences=1,
+        do_sample=True,
+        top_k=50,
+        temperature=0.7,
+    )[0]["generated_text"]
     return response
 
+def commit_and_push_changes(commit_message):
+    """Commits and pushes changes to the Hugging Face repository."""
+    try:
+        os.system(f"git add .")
+        os.system(f"git commit -m '{commit_message}'")
+        os.system(f"git push origin main")
+        st.write(f"Changes committed and pushed to {HUGGING_FACE_REPO_URL}")
+    except Exception as e:
+        st.write(f"Error committing and pushing changes: {e}")
+
+def format_code(code):
+    """Formats the code using Black and checks for linting errors."""
+    try:
+        formatted_code = black.format_str(code, mode=black.FileMode())
+        return formatted_code
+    except Exception as e:
+        return f"Error formatting code: {e}"
+
+def lint_code(code):
+    """Checks the code for linting errors using Pylint."""
+    try:
+        output = StringIO()
+        lint.run(code, do_exit=False, output=output)
+        return output.getvalue()
+    except Exception as e:
+        return f"Error linting code: {e}"
+
 def workspace_interface(project_name):
     project_path = os.path.join(PROJECT_ROOT, 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": []}
-        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."
+    st.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."
     else:
         return f"Project {project_name} already exists."
 
@@ -166,25 +230,39 @@ def sentiment_analysis(text):
     return sentiment[0]
 
 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",
+        "Go": ".go",
+        "Ruby": ".rb",
+        "PHP": ".php",
+        "Swift": ".swift",
+        "Kotlin": ".kt",
+        "TypeScript": ".ts",
+        "Rust": ".rs",
+        "Scala": ".scala",
+        "Dart": ".dart",
+        "Lua": ".lua",
+        "HTML": ".html",
+        "CSS": ".css",
+        "SQL": ".sql"
     }
 
-    # Add code to handle edge cases such as invalid input and unsupported programming languages
     if input_language not in language_extensions:
         raise ValueError(f"Invalid input language: {input_language}")
     if output_language not in language_extensions:
         raise ValueError(f"Invalid output language: {output_language}")
 
-    # Use the dictionary to map the input and output languages to their corresponding file extensions
     input_extension = language_extensions[input_language]
     output_extension = language_extensions[output_language]
 
     # Translate the code using the OpenAI API
     prompt = f"Translate this code from {input_language} to {output_language}:\n\n{code}"
     response = openai.ChatCompletion.create(
-        model="gpt-4",
+        model="gpt-3.5-turbo",
         messages=[
             {"role": "system", "content": "You are an expert software developer."},
             {"role": "user", "content": prompt}
@@ -193,7 +271,6 @@ def translate_code(code, input_language, output_language):
     translated_code = response.choices[0].message['content'].strip()
 
     # Return the translated code
-    translated_code = response.choices[0].message['content'].strip()
     st.session_state.current_state['toolbox']['translated_code'] = translated_code
     return translated_code
 
@@ -203,11 +280,10 @@ def generate_code(code_idea):
     generator = pipeline('text-generation', model='gpt4o')
     generated_code = generator(code_idea, max_length=10000, num_return_sequences=1)[0]['generated_text']
     messages=[
-            {"role": "system", "content": "You are an expert software developer."},
-            {"role": "user", "content": f"Generate a Python code snippet for the following idea:\n\n{code_idea}"}
-        ]
+        {"role": "system", "content": "You are an expert software developer."},
+        {"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 run_git_command(command):
@@ -247,23 +323,22 @@ if scope == "Local" and not is_git_repository():
 
 if st.button("Set Git Configuration"):
     global_flag = "--global" if scope == "Global" else ""
-    
+
     email_command = f"git config {global_flag} user.email \"{email}\""
     name_command = f"git config {global_flag} user.name \"{name}\""
-    
+
     success, email_result = run_git_command(email_command)
     if success:
         st.success("Email configuration set successfully!")
     else:
         st.error(f"Failed to set email configuration: {email_result}")
-    
+
     success, name_result = run_git_command(name_command)
     if success:
         st.success("Name configuration set successfully!")
     else:
         st.error(f"Failed to set name configuration: {name_result}")
-
-st.write("---")
+    st.write("---")
 
 if st.button("View Current Git Configuration"):
     success, config_result = run_git_command("git config --list")
@@ -273,16 +348,6 @@ if st.button("View Current Git Configuration"):
     else:
         st.error(f"Failed to retrieve Git configuration: {config_result}")
 
-def commit_and_push_changes(commit_message):
-    """Commits and pushes changes to the Hugging Face repository."""
-    try:
-        subprocess.run(["git", "add", "."], check=True)
-        subprocess.run(["git", "commit", "-m", commit_message], check=True)
-        subprocess.run(["git", "push", "origin", "main"], check=True)
-        return True, "Changes committed and pushed successfully."
-    except subprocess.CalledProcessError as e:
-        return False, f"Error in Git operations: {e}"
-
 # Streamlit App
 st.title("AI Agent Creator")
 
@@ -331,7 +396,8 @@ elif app_mode == "Tool Box":
     st.subheader("Code Editor")
     code_editor = st.text_area("Write your code:", height=300)
     if st.button("Format & Lint"):
-        formatted_code, lint_message = code_editor_interface(code_editor)
+        formatted_code = format_code(code_editor)
+        lint_message = lint_code(code_editor)
         st.code(formatted_code, language="python")
         st.info(lint_message)
 
@@ -377,7 +443,7 @@ elif app_mode == "Tool Box":
         "Translate code": "translate_code('code', 'source_language', 'target_language')",
     }
     for command_name, command in preset_commands.items():
-        st.write(f"{command_name}: `{command}`")
+        st.write(f"{command_name}: {command}")
 
 elif app_mode == "Workspace Chat App":
     # Workspace Chat App
@@ -430,7 +496,7 @@ elif app_mode == "Workspace Chat App":
     for project, details in st.session_state.workspace_projects.items():
         st.write(f"Project: {project}")
         for file in details['files']:
-            st.write(f"  - {file}")
+            st.write(f" - {file}")
 
     # Chat with AI Agents
     st.subheader("Chat with AI Agents")
@@ -451,5 +517,54 @@ elif app_mode == "Workspace Chat App":
         st.write("Next Step:")
         st.write(next_step)
 
-# Display current state for debugging
-st.sidebar.subheader("Current State")
+    # Display current state for debugging
+    st.sidebar.subheader("Current State")
+    st.sidebar.write(st.session_state.current_state)
+
+def show_chat_history(current_agent):
+    """Displays the chat history."""
+    if current_agent in st.session_state.current_state['workspace_chat']:
+        chat_msgs = st.session_state.current_state['workspace_chat'][current_agent]['messages']
+        for msg_input, _ in reversed(chat_msgs):
+            st.markdown(f"<s>{current_agent}: {msg_input}</s>", unsafe_allow_html=True)
+    return ""
+
+def get_agent_response(agent, input_text):
+    """Gets the response from an agent for a given input."""
+    if agent is None:
+        return "No agent selected."
+    elif agent == "CodeCraft":
+        return chat_interface(input_text)
+    else:
+        return chat_interface_with_agent(input_text, agent)
+
+def send_message_to_agent(agent, input_text):
+    """Sends a message to an agent and displays its response."""
+    output_text = get_agent_response(agent, input_text)
+    st.markdown(f"{agent}: {output_text}")
+
+def process_chat_input(input_text, current_agent):
+    """Processes a chat input by checking special characters and forwarding it to appropriate handlers."""
+    if re.match(r'^\W+$', input_text): # Mention other agents
+        words = input_text[1:].split(' ')
+        mentioned_agents = [' '.join(words[:i]+['']) for i in range(len(words)+1)]
+        if len(mentioned_agents) > 1:
+            del mentioned_agents[-1]
+        mention_msg = f"@{current_agent}, you mentioned: {' | '.join(mentioned_agents)}"
+        st.markdown(mention_msg)
+        for agent in mentioned_agents:
+            if agent != '' and agent in st.session_state.available_agents:
+                st.session_state.current_state['workspace_chat'][agent] = {'messages': [], 'active': True}
+                st.session_state.current_state['workspace_chat'][current_agent]['messages'].append((input_text, ""))
+    elif re.fullmatch(r'\d+', input_text): # Select agent by number
+        if int(input_text)-1 < len(st.session_state.available_agents):
+            st.session_state.current_agent = st.session_state.available_agents[int(input_text)-1]
+    elif input_text == '/clear': # Clear conversation
+        st.session_state.current_state['workspace_chat'][current_agent]['messages'] = []
+    elif input_text == '/exit': # Exit conversation
+        st.session_state.current_agent = None
+    else:
+        st.session_state.current_state['workspace_chat'][current_agent]['messages'].append((input_text, ""))
+
+if __name__ == "__main__":
+    main()