Update app2.py

app2.py

@@ -1,41 +1,15 @@
-import asyncio
-import gradio as gr
-from sqlalchemy.exc import SQLAlchemyError
-from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession
-from sqlalchemy.future import select  # Correct async query API
-from sqlalchemy.orm import sessionmaker
-import logging
-import os
-import sys
-import subprocess
-from transformers import pipeline, AutoModelForSeq2SeqLM, AutoTokenizer
-import openai
 import streamlit as st
-from io import StringIO
-from rich import print as rprint
-from rich.panel import Panel
-from rich.progress import track
-from rich.table import Table
-import git
-from langchain.llms import HuggingFaceHub
+from langchain.llms import GoogleSearchAPIWrapper
 from langchain.chains import ConversationChain
 from langchain.memory import ConversationBufferMemory
+import subprocess
+import git
 
-# Constants
-MODEL_NAME = "google/flan-t5-xl"
-MAX_NEW_TOKENS = 2048
-TEMPERATURE = 0.7
-TOP_P = 0.95
-REPETITION_PENALTY = 1.2
-
-# Load Model and Tokenizer
-@st.cache_resource
-def load_model_and_tokenizer():
-    model = AutoModelForSeq2SeqLM.from_pretrained(MODEL_NAME, device_map="auto")
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-    return model, tokenizer
+# Replace with your actual Google Search API key
+GOOGLE_API_KEY = "YOUR_API_KEY"  
 
-model, tokenizer = load_model_and_tokenizer()
+# Initialize Google Search API Wrapper
+search = GoogleSearchAPIWrapper(google_api_key=GOOGLE_API_KEY)
 
 # Agents
 agents = {
@@ -82,14 +56,20 @@ def add_code_to_workspace(project_name: str, code: str, file_name: str):
         return f"Project {project_name} does not exist"
 
 def terminal_interface(command: str, project_name: str):
-    if project_name in st.session_state.workspace_projects:
-        result = subprocess.run(command, cwd=project_name, shell=True, capture_output=True, text=True)
-        return result.stdout + result.stderr
-    else:
-        return f"Project {project_name} does not exist"
+    try:
+        if project_name in st.session_state.workspace_projects:
+            result = subprocess.run(command, cwd=project_name, shell=True, capture_output=True, text=True)
+            return result.stdout + result.stderr
+        else:
+            return f"Project {project_name} does not exist"
+    except FileNotFoundError:
+        return f"Error: Command not found. Please check your command."
+    except Exception as e:
+        logging.error(f"An error occurred: {e}")
+        return f"An unexpected error occurred while running the command."
 
 def get_agent_response(message: str, system_prompt: str):
-    llm = HuggingFaceHub(repo_id=MODEL_NAME, model_kwargs={"temperature": TEMPERATURE, "top_p": TOP_P, "repetition_penalty": REPETITION_PENALTY, "max_length": MAX_NEW_TOKENS})
+    llm = GoogleSearchAPIWrapper(google_api_key=GOOGLE_API_KEY)
     memory = ConversationBufferMemory()
     conversation = ConversationChain(llm=llm, memory=memory)
     response = conversation.run(system_prompt + "\n" + message)
@@ -114,7 +94,7 @@ def display_chat_history():
     for message in st.session_state.chat_history:
         st.text(message)
 
-def run_autonomous_build(selected_agents: List[str], project_name: str):
+def run_autonomous_build(selected_agents: list[str], project_name: str):
     st.info("Starting autonomous build process...")
     for agent in selected_agents:
         st.write(f"Agent {agent} is working on the project...")
@@ -123,7 +103,7 @@ def run_autonomous_build(selected_agents: List[str], project_name: str):
         st.write(f"Agent {agent} has completed its task.")
     st.success("Autonomous build process completed!")
 
-def collaborative_agent_example(selected_agents: List[str], project_name: str, task: str):
+def collaborative_agent_example(selected_agents: list[str], project_name: str, task: str):
     st.info(f"Starting collaborative task: {task}")
     responses = {}
     for agent in selected_agents:
@@ -135,7 +115,7 @@ def collaborative_agent_example(selected_agents: List[str], project_name: str, t
     st.success("Collaborative task completed!")
     st.write(combined_response)
 
-def combine_and_process_responses(responses: Dict[str, str], task: str) -> str:
+def combine_and_process_responses(responses: dict[str, str], task: str) -> str:
     combined = "\n\n".join([f"{agent}: {response}" for agent, response in responses.items()])
     return f"Combined response for task '{task}':\n\n{combined}"
 
@@ -163,7 +143,7 @@ if st.session_state.workspace_projects:
         selected_file = st.selectbox("Select file to edit", files) if files else None
         if selected_file:
             file_content = next((file['code'] for file in st.session_state.workspace_projects[selected_project]['files'] if file['file_name'] == selected_file), "")
-            edited_code = st_ace(value=file_content, language="python", theme="monokai", key="code_editor")
+            edited_code = st.text_area("Edit code", value=file_content, height=300)  # Using st.text_area for now
             if st.button("Save Changes"):
                 for file in st.session_state.workspace_projects[selected_project]['files']:
                     if file['file_name'] == selected_file:
@@ -173,7 +153,7 @@ if st.session_state.workspace_projects:
         else:
             st.info("No files in the project. Use the chat interface to generate code.")
 else:
-    st.info("No projects created yet. Create a project to start coding.")
+    st.info(   "No projects created yet. Create a project to start coding."
 
 # Terminal Interface
 st.subheader("Terminal (Workspace Context)")
@@ -246,4 +226,4 @@ if st.button("Start Collaborative Task"):
     if collab_agents and collab_project and collab_task:
         collaborative_agent_example(collab_agents, collab_project, collab_task)
     else:
-        st.warning("Please select agents, enter a project name, and a task.")
+        st.warning("Please select agents, enter a project name, and a task.")