import os
import gradio as gr
import tempfile
import zipfile
from groq import Groq
from dotenv import load_dotenv
from .generator import AIReadmeGenerator

# Validate the API key using the provided format
def validate_api_key(api_key):
    """
    Validate the API key using a test completion request.
    """
    try:
        client = Groq(api_key=api_key)
        response = client.chat.completions.create(
            model="llama-3.1-8b-instant",
            messages=[
                {"role": "system", "content": "Validation test for API key."},
                {"role": "user", "content": "Hello, world!"}
            ],
            temperature=0.1,
            max_tokens=5,
        )
        if response and response.choices:
            return True, "API key is valid!"
        else:
            return False, "API key validation failed: Empty response."
    except Exception as e:
        return False, f"Invalid API key. Error: {str(e)}"

# Generate README
def generate_readme_with_key(project_folder, api_key):
    """
    Generate the README for a given project folder using the provided API key.
    """
    try:
        # Use the API key dynamically
        generator = AIReadmeGenerator(api_key)
        with tempfile.TemporaryDirectory() as temp_dir:
            # Extract uploaded ZIP file
            with zipfile.ZipFile(project_folder, 'r') as zip_ref:
                zip_ref.extractall(temp_dir)
            
            # Generate the README
            readme_content = generator.generate_concise_readme(temp_dir)
            
            return readme_content, readme_content
    except zipfile.BadZipFile:
        return "Error: Uploaded file is not a valid ZIP archive.", "Error: Invalid ZIP archive."
    except Exception as e:
        return f"Error generating README: {str(e)}", f"Error generating README: {str(e)}"

# Handle user login
def on_login(api_key):
    is_valid, message = validate_api_key(api_key)
    if is_valid:
        return gr.update(visible=False), gr.update(visible=True), gr.Markdown(f"**Success!** {message}", visible=True)
    else:
        return gr.update(visible=True), gr.update(visible=False), gr.Markdown(f"**Error!** {message}", visible=True)

# Create Gradio UI
def create_ui():
    with gr.Blocks(title="AI README Generator") as demo:
        # Login Section
        with gr.Column(visible=True) as login_section:
            gr.Markdown("## Login to Access the AI README Generator")
            api_key_input = gr.Textbox(label="Enter your API Key:", type="password")
            login_button = gr.Button("Login")
            login_feedback = gr.Markdown(visible=False)

            gr.Markdown("""
            ### Login Instructions
            1. **Sign Up / Log In** at [Groq API Portal](https://console.groq.com/keys).
            2. **Generate an API Key** in the API Keys section.
            3. **Copy and Paste the API Key** into the box above.
            4. **Click Login** to proceed.
            """)

        # README Generator Section
        with gr.Column(visible=False) as readme_section:
            gr.Markdown("# AI README Craft")
            gr.Markdown("""
            ## Instructions
            - **Step 1:** Convert your project folder into a ZIP file.
            - **Step 2:** Upload the ZIP file using the form below.
            - **Step 3:** Click the **Generate README** button.
            - **Note:** Larger files may take more time to process. Please be patient!
            """)
            folder_input = gr.File(file_count="single", type="filepath", label="📂 Upload Your Project (ZIP)")
            generate_btn = gr.Button("✨ Generate README")
            with gr.Row():
                readme_output = gr.Textbox(lines=20, interactive=False, label="Generated README")
                readme_markdown = gr.Markdown(label="README Preview")
        
        # Button Click Events
        login_button.click(
            on_login,
            inputs=[api_key_input],
            outputs=[login_section, readme_section, login_feedback]
        )
        generate_btn.click(
            generate_readme_with_key,
            inputs=[folder_input, api_key_input],
            outputs=[readme_output, readme_markdown]
        )

    return demo