utils.py

import json
from google.genai import Client
from keys import GEMINI_API_KEY
import os


client = Client(api_key = GEMINI_API_KEY)


def get_project_code_from_path(project_path):
    project_code_list = []
    skip_dirs = [".venv", "__pycache__"]

    for root, directories, files in os.walk(project_path):
        directories[:] = [name for name in directories if name not in skip_dirs]

        for f in files:
            if not f.endswith((".py",".js",".ts",".go",".java",".txt")):
                continue

            path = os.path.join(root, f)
            with open(path, "r") as file:
                project_code_list.append({
                    "file": path,
                    "content": file.read()
                })

        return project_code_list


def save_project_code_list(project_code_list):
    try:
        with open('./project_code/project_code_list.json', 'w') as file:
            json.dump(project_code_list, file, indent = 4)
    except Exception as e:
        print(f"Error saving project code list: {e}")


def load_project_code_list():
    try:
        with open('./project_code/project_code_list.json', 'r') as file:
            return json.load(file)
    except Exception as e:
        print(f"Error loading project code list: {e}")
        return []


def invoke_gemini(gemini_model, model_prompt):
    response = client.models.generate_content(
        model = gemini_model,
        contents = f'{model_prompt}'
    ).text
    return response