Blame for app.py

import google.generativeai as genai
import gradio as gr
import yaml
from datetime import datetime
import uuid
import sys
import traceback

# Print version info for debugging
print(f"Python version: {sys.version}")
print(f"Gradio version: {gr.__version__}")

# Set up the model
generation_config = {
    "temperature": 0.2,
    "top_p": 1,
    "top_k": 32,
    "max_output_tokens": 8192,
}

safety_settings = [
    {
        "category": "HARM_CATEGORY_HARASSMENT",
        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
    },
    {
        "category": "HARM_CATEGORY_HATE_SPEECH",
        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
    },
    {
        "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
    },
    {
        "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
        "threshold": "BLOCK_MEDIUM_AND_ABOVE"
    }
]

# Note: Replace with your actual API key
genai.configure(api_key="AIzaSyBF8c8pxDa8Y6pUWcU-V9vvlFBDER4bY2E")
model = genai.GenerativeModel(
    model_name="gemini-2.0-flash",
    generation_config=generation_config,
    safety_settings=safety_settings
)

def generate_changeset_id():
    """Generate a unique changeset ID with timestamp"""
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    return f"changeset_{timestamp}_{str(uuid.uuid4())[:8]}"

def generate_liquibase_changeset(user_prompt, existing_changelog, author_name, context_env):
    """Generate a new changeset based on user description"""
    
    changeset_id = generate_changeset_id()
    
    input_prompt = f"""
You are a Liquibase expert. Generate a new changeset in YAML format based on the user's description.

USER REQUEST: {user_prompt}

EXISTING CHANGELOG:
{existing_changelog if existing_changelog and existing_changelog.strip() else "No existing changelog provided."}

AUTHOR: {author_name}
CONTEXT: {context_env}

Please generate a complete Liquibase changeset that:
1. Has a unique changeset ID: {changeset_id}
2. Includes proper rollback statements
3. Uses appropriate data types for the database
4. Includes proper constraints and indexes where needed
5. Follows Liquibase best practices
6. Is properly formatted YAML

Respond with ONLY the changeset YAML (not the full databaseChangeLog wrapper), starting with:

- changeSet:
    id: {changeset_id}
    author: {author_name}
    context: {context_env}
    comment: [Generated comment describing the change]
    changes:
      [Your generated changes here]
    rollback:
      [Your rollback statements here]

Examples of common operations:
- Add column: Use addColumn with proper type and constraints
- Drop column: Use dropColumn with rollback addColumn
- Create table: Use createTable with all columns and constraints
- Add index: Use createIndex with rollback dropIndex
- Modify column: Use modifyDataType or addNotNullConstraint
- Add foreign key: Use addForeignKeyConstraint

Make sure the changeset is production-ready and includes appropriate error handling.
"""

    try:
        response = model.generate_content(input_prompt)
        return response.text
    except Exception as e:
        return f"Error generating changeset: {str(e)}"

def process_changeset_request(user_prompt, existing_changelog, author_name, context_env):
    """Main function to process user request and generate changeset"""
    
    try:
        if not user_prompt or not user_prompt.strip():
            return "Please provide a description of the database change you want to make."
        
        # Set defaults
        if not author_name or not author_name.strip():
            author_name = "harness"
        if not context_env or not context_env.strip():
            context_env = "dev"
        
        # Generate the changeset
        response = generate_liquibase_changeset(user_prompt, existing_changelog, author_name, context_env)
        
        return response
        
    except Exception as e:
        print(f"Error in process_changeset_request: {e}")
        traceback.print_exc()
        return f"Error processing request: {str(e)}"

# Alternative interface creation with explicit types
def create_interface_alternative():
    """Alternative interface creation method"""
    try:
        # Create inputs with explicit typing
        inputs = [
            gr.components.Textbox(
                placeholder="e.g., 'Add a column named email to the users table'",
                label="Describe Your Database Change",
                lines=3
            ),
            gr.components.Textbox(
                placeholder="Paste your current changelog.yml content here (optional)",
                label="Existing Changelog Content (Optional)",
                lines=5
            ),
            gr.components.Textbox(
                placeholder="your-username",
                label="Author Name",
                value="harness"
            ),
            gr.components.Textbox(
                placeholder="dev, test, prod",
                label="Context/Environment",
                value="dev"
            ),
        ]
        
        outputs = gr.components.Textbox(
            label="Generated Changeset",
            lines=20,
            show_copy_button=True
        )
        
        interface = gr.Interface(
            fn=process_changeset_request,
            inputs=inputs,
            outputs=outputs,
            title="🔄 Harness Database DevOps Changelog Generator",
            description="Describe your database change in plain English, and get a production-ready changelog",
            article="⚡ Powered by <div style='text-align: center; margin-top: 20px; padding: 10px; border-top: 1px solid #e0e0e0;'><a href='https://developer.harness.io/docs/category/get-started-with-database-devops' target='_blank' style='color: #0066cc; text-decoration: none; font-weight: 500;'>Harness Database DevOps</a></div>",
            examples=[
                [
                    "Add a column named 'email' of type VARCHAR(255) to the users table",
                    "",
                    "developer",
                    "dev"
                ],
                [
                    "Create a new table called 'orders' with id, user_id, product_id, and created_at columns",
                    "",
                    "developer",
                    "dev"
                ],
                [
                    "Create an index on the email column in the users table",
                    "",
                    "developer",
                    "dev"
                ],
            ],
            allow_flagging="never"
        )
        return interface
    except Exception as e:
        print(f"Error creating alternative interface: {e}")
        traceback.print_exc()
        return None

def create_interface():
    """Original interface creation method with error handling"""
    try:
        interface = gr.Interface(
            fn=process_changeset_request,
            inputs=[
                gr.Textbox(
                    placeholder="e.g., 'Add a column named email to the users table'",
                    label="Describe Your Database Change",
                    lines=3
                ),
                gr.Textbox(
                    placeholder="Paste your current changelog.yml content here (optional)",
                    label="Existing Changelog Content (Optional)",
                    lines=5
                ),
                gr.Textbox(
                    placeholder="your-username",
                    label="Author Name",
                    value="harness dbops"
                ),
                gr.Textbox(
                    placeholder="dev, test, prod",
                    label="Context/Environment",
                    value="dev"
                ),
            ],
            outputs=gr.Textbox(
                label="Generated Changeset",
                lines=20,
                show_copy_button=True
            ),
            title="🔄 Harness Database DevOps Changelog Generator",
            description="Describe your database change in plain English, and get a production-ready changelog",
            article="⚡ Powered by <div style='text-align: center; margin-top: 20px; padding: 10px; border-top: 1px solid #e0e0e0;'><a href='https://developer.harness.io/docs/category/get-started-with-database-devops' target='_blank' style='color: #0066cc; text-decoration: none; font-weight: 500;'>Harness Database DevOps</a></div>",
            examples=[
                [
                    "Add a column named 'email' of type VARCHAR(255) to the users table",
                    "",
                    "developer",
                    "dev"
                ],
                [
                    "Create a new table called 'orders' with id, user_id, product_id, and created_at columns",
                    "",
                    "developer",
                    "dev"
                ],
                [
                    "Create an index on the email column in the users table",
                    "",
                    "harness",
                    "dev"
                ],
            ],
            allow_flagging="never"
        )
        return interface
    except Exception as e:
        print(f"Error creating original interface: {e}")
        traceback.print_exc()
        return create_interface_alternative()

if __name__ == "__main__":
    try:
        demo = create_interface()
        if demo is not None:
            demo.launch(
                debug=True,
                share=True,
                server_name="0.0.0.0",
                server_port=7860
            )
        else:
            print("Failed to create interface")
    except Exception as e:
        print(f"Error launching application: {e}")
        traceback.print_exc()