Update app.py

app.py

@@ -12,11 +12,6 @@ from i_search import google, i_search as i_s
 from transformers import AutoModelForCausalLM, AutoTokenizer
 import random
 import prompts
-client = InferenceClient(
-    "mistralai/Mixtral-8x7B-Instruct-v0.1"
-)
-
-from langchain_community.embeddings import HuggingFaceEmbeddings
 
 # --- Configuration ---
 VERBOSE = True
@@ -25,7 +20,7 @@ MAX_TOKENS = 2048
 TEMPERATURE = 0.7
 TOP_P = 0.8
 REPETITION_PENALTY = 1.5
-MODEL_NAME = "codellama/CodeLlama-13b-Python-hf"
+MODEL_NAME = "codellama/CodeLlama-13b-Python-hf"  # Use CodeLlama for code-related tasks
 API_KEY = os.getenv("HUGGINGFACE_API_KEY")
 
 # --- Logging Setup ---
@@ -161,8 +156,8 @@ def mixtral_generate(
     repetition_penalty: float = REPETITION_PENALTY,
 ) -> str:
     """Generates a response using the Mixtral model."""
-    tokenizer = AutoTokenizer.from_pretrained(MODEL_NAME)
-    model = AutoModelForCausalLM.from_pretrained(MODEL_NAME)
+    tokenizer = AutoTokenizer.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")  # Use Mixtral model
+    model = AutoModelForCausalLM.from_pretrained("mistralai/Mixtral-8x7B-Instruct-v0.1")
 
     content = PREFIX.format(
         date_time_str=datetime.now().strftime("%Y-%m-%d %H:%M:%S"),
@@ -362,6 +357,7 @@ def main():
         def create_file(project_path: str, file_name: str) -> str:
             """Creates a new file in the project directory."""
             try:
+                os.makedirs(os.path.dirname(os.path.join(project_path, file_name)), exist_ok=True)  # Create directory if needed
                 open(os.path.join(project_path, file_name), "a").close()
                 return f"File {file_name} created successfully."
             except Exception as e: