import os
from dotenv import load_dotenv
from openai import AzureOpenAI

load_dotenv()

client = AzureOpenAI(
    api_version="2025-01-01-preview",
    azure_endpoint=os.getenv("AZURE_OPENAI_ENDPOINT").strip(),
    api_key=os.getenv("AZURE_OPENAI_KEY").strip()
)

def enhance_with_ai(repo_info):
    file_summaries = "\n".join([
        f"### {name}\n```{content[:300]}```"
        for name, content in list(repo_info['files'].items())[:10]
    ])

    prompt = f"""
    Create a professional README.md for a GitHub project named "{repo_info['name']}".

    Description: {repo_info['description']}
    Topics: {', '.join(repo_info['topics'])}

    The project includes the following files:
    {file_summaries}

    README must include: Introduction, Features, Installation, Usage, Contributing, and License.
    """

    response = client.chat.completions.create(
        model=os.getenv("AZURE_DEPLOYMENT_NAME").strip(),
        messages=[
            {"role": "system", "content": "You are a helpful assistant that creates GitHub READMEs. You must follow the provided structure and include all necessary sections. Use necessary emojies and markdown formatting."},
            {"role": "user", "content": prompt}
        ],
        temperature=0.7,
        max_tokens=1000
    )

    return response.choices[0].message.content