app.py

import gradio as gr
import logging
from services.tmdb_client import tmdb_client
from utils.config import config

# Configure logging
logging.basicConfig(
    level=logging.INFO,
    format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
    handlers=[
        logging.FileHandler('movie_app.log'),
        logging.StreamHandler()
    ]
)

# Create logger for this module
logger = logging.getLogger(__name__)

def extract_country_code(country_selection):
    """Extract country code from selection like 'United States (US)' -> 'US'"""
    if not country_selection:
        return "US"  # Default fallback
    
    # Extract code from parentheses
    if "(" in country_selection and ")" in country_selection:
        return country_selection.split("(")[-1].replace(")", "").strip()
    
    # Fallback if format is unexpected
    return "US"

def get_movie_recommendations(decade, country, genre, num_movies):
    """Get movie recommendations based on user preferences."""
    logger.info(f"Recommendation request: decade={decade}, country={country}, genre={genre}, num_movies={num_movies}")
    
    if not decade or not country:
        logger.warning("Missing required parameters: decade or country")
        return "Please select decade and country."
    
    # Check if country requires genre selection
    if country in config.TOP_COUNTRIES and not genre:
        logger.warning(f"Genre required for top country {country} but not provided")
        return "Please select a genre for this country."
    
    # Convert decade string to integer (e.g., "2000s" -> 2000)
    decade_year = int(decade.replace("s", ""))
    
    # Extract country code from the selection
    country_code = extract_country_code(country)
    
    # Get recommendations (genre is optional for non-top countries)
    logger.info(f"Fetching movies for {country_code}, {decade_year}, genre: {genre}")
    movies = tmdb_client.get_random_movies(
        decade=decade_year,
        country=country_code,
        genre=genre if country in config.TOP_COUNTRIES else None,
        n=num_movies
    )
    
    # Update result message based on country type
    if country in config.TOP_COUNTRIES:
        search_criteria = f"{genre} from {country} ({decade})"
    else:
        search_criteria = f"from {country} ({decade})"
    
    if not movies:
        logger.warning(f"No movies found {search_criteria}")
        return f"No movies found {search_criteria}."
    
    logger.info(f"Successfully found {len(movies)} movies {search_criteria}")
    
    # Format results with mobile-friendly movie information
    output = f"🎬 **{len(movies)} movies {search_criteria}:**\n\n"
    
    for i, movie in enumerate(movies, 1):
        # Format title - show original title if different from English title
        title_display = movie['title']
        if movie.get('original_title') and movie['original_title'] != movie['title']:
            title_display = f"{movie['title']} ({movie['original_title']})"
        
        # Create a mobile-friendly vertical card layout
        output += f"<div style='margin-bottom: 20px; padding: 15px; border: 2px solid #e0e0e0; border-radius: 12px; background: linear-gradient(145deg, #ffffff, #f5f5f5); box-shadow: 0 6px 12px rgba(0,0,0,0.1); max-width: 100%;'>\n"
        
        # Movie title header
        output += f"<h2 style='margin: 0 0 15px 0; color: #2c3e50; font-size: 1.3em; text-align: center;'>{i}. {title_display}</h2>\n"
        
        # Poster image section - centered and smaller
        if movie.get('poster_url'):
            output += f"<div style='display: flex; justify-content: center; align-items: center; margin: 20px 0; padding: 10px;'>\n"
            output += f"<img src='{movie['poster_url']}' style='width: 130px; height: 195px; object-fit: cover; border-radius: 10px; box-shadow: 0 4px 8px rgba(0,0,0,0.2);' alt='{movie['title']} poster'>\n"
            output += f"</div>\n"
        
        # Basic Info Row - more compact
        output += f"<div style='display: flex; flex-wrap: wrap; gap: 10px; margin-bottom: 12px; justify-content: center;'>\n"
        output += f"<div style='background: #ecf0f1; padding: 6px 10px; border-radius: 6px; font-size: 0.9em;'><strong>Year:</strong> {movie['year']}</div>\n"
        
        if movie.get('has_omdb_data') and movie.get('runtime') != 'N/A':
            output += f"<div style='background: #ecf0f1; padding: 6px 10px; border-radius: 6px; font-size: 0.9em;'><strong>Runtime:</strong> {movie['runtime']}</div>\n"
        
        if movie.get('has_omdb_data') and movie.get('rated') != 'N/A':
            output += f"<div style='background: #ecf0f1; padding: 6px 10px; border-radius: 6px; font-size: 0.9em;'><strong>Rated:</strong> {movie['rated']}</div>\n"
        
        output += f"</div>\n"
        
        # Director and Cast (if available) - more compact
        if movie.get('has_omdb_data'):
            if movie.get('director') != 'N/A':
                output += f"<p style='margin: 6px 0; color: #34495e; font-size: 0.95em; text-align: center;'><strong>🎬 Director:</strong> {movie['director']}</p>\n"
            
            if movie.get('actors') != 'N/A':
                actors = movie['actors'][:120] + '...' if len(movie['actors']) > 120 else movie['actors']
                output += f"<p style='margin: 6px 0; color: #34495e; font-size: 0.95em; text-align: center;'><strong>🎭 Cast:</strong> {actors}</p>\n"
        
        # Ratings Section - 2x2 grid layout for mobile
        output += f"<div style='background: #f8f9fa; padding: 12px; border-radius: 8px; margin: 12px 0;'>\n"
        output += f"<h4 style='margin: 0 0 10px 0; color: #2c3e50; text-align: center;'>📊 Ratings</h4>\n"
        output += f"<div style='display: grid; grid-template-columns: 1fr 1fr; gap: 8px; max-width: 300px; margin: 0 auto;'>\n"
        
        # TMDB Rating
        output += f"<div style='background: #3498db; color: white; padding: 8px; border-radius: 6px; text-align: center;'>\n"
        output += f"<div style='font-weight: bold; font-size: 0.9em;'>TMDB</div>\n"
        output += f"<div style='font-size: 0.9em;'>⭐ {movie['rating']}/10</div>\n"
        output += f"<div style='font-size: 0.7em;'>({movie['vote_count']} votes)</div>\n"
        output += f"</div>\n"
        
        # IMDB Rating (if available)
        if movie.get('has_omdb_data') and movie.get('imdb_rating') != 'N/A':
            output += f"<div style='background: #f39c12; color: white; padding: 8px; border-radius: 6px; text-align: center;'>\n"
            output += f"<div style='font-weight: bold; font-size: 0.9em;'>IMDb</div>\n"
            output += f"<div style='font-size: 0.9em;'>⭐ {movie['imdb_rating']}/10</div>\n"
            if movie.get('imdb_votes') != 'N/A':
                output += f"<div style='font-size: 0.7em;'>({movie['imdb_votes']} votes)</div>\n"
            output += f"</div>\n"
        
        # Rotten Tomatoes (if available)
        if movie.get('has_omdb_data') and movie.get('rotten_tomatoes') != 'N/A':
            output += f"<div style='background: #e74c3c; color: white; padding: 8px; border-radius: 6px; text-align: center;'>\n"
            output += f"<div style='font-weight: bold; font-size: 0.9em;'>RT</div>\n"
            output += f"<div style='font-size: 0.9em;'>🍅 {movie.get('rotten_tomatoes')}</div>\n"
            output += f"</div>\n"
        
        # Metacritic (if available)
        if movie.get('has_omdb_data') and movie.get('metascore') != 'N/A':
            output += f"<div style='background: #9b59b6; color: white; padding: 8px; border-radius: 6px; text-align: center;'>\n"
            output += f"<div style='font-weight: bold; font-size: 0.9em;'>Meta</div>\n"
            output += f"<div style='font-size: 0.9em;'>📊 {movie['metascore']}/100</div>\n"
            output += f"</div>\n"
        
        output += f"</div>\n"
        output += f"</div>\n"
        
        # Plot/Overview - more compact
        plot_text = movie.get('plot', movie['overview']) if movie.get('has_omdb_data') else movie['overview']
        plot_display = plot_text[:250] + '...' if len(plot_text) > 250 else plot_text
        output += f"<p style='margin: 12px 0; color: #555; line-height: 1.4; background: #fff; padding: 10px; border-radius: 6px; border-left: 3px solid #3498db; font-size: 0.95em;'><strong>📝 Plot:</strong> {plot_display}</p>\n"
        
        # Awards (if available) - more compact
        if movie.get('has_omdb_data') and movie.get('awards') != 'N/A' and movie['awards'] != 'No awards.':
            awards_text = movie['awards'][:150] + '...' if len(movie['awards']) > 150 else movie['awards']
            output += f"<p style='margin: 8px 0; color: #e67e22; background: #fef9e7; padding: 8px; border-radius: 6px; border-left: 3px solid #f39c12; font-size: 0.9em;'><strong>🏆 Awards:</strong> {awards_text}</p>\n"
        
        # Streaming Services Section - more compact
        streaming_services = movie.get('all_services', [])
        if streaming_services:
            output += f"<div style='background: #e8f5e8; padding: 12px; border-radius: 8px; margin: 12px 0; border: 2px solid #27ae60;'>\n"
            output += f"<h4 style='margin: 0 0 10px 0; color: #27ae60; display: flex; align-items: center; justify-content: center; font-size: 1em;'><span style='margin-right: 6px;'>📺</span>Watch Now</h4>\n"
            
            # Free services
            free_services = movie.get('free_services', [])
            if free_services:
                output += f"<div style='margin-bottom: 10px;'>\n"
                output += f"<div style='font-weight: bold; color: #2c5530; margin-bottom: 6px; font-size: 0.9em; text-align: center;'>🆓 Free</div>\n"
                output += f"<div style='display: flex; flex-wrap: wrap; gap: 6px; justify-content: center;'>\n"
                for service in free_services:
                    if service.get('logo_url'):
                        output += f"<div style='display: flex; align-items: center; background: white; padding: 4px 8px; border-radius: 6px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); border: 1px solid #ddd;'>\n"
                        output += f"<img src='{service['logo_url']}' style='width: 20px; height: 20px; margin-right: 4px; border-radius: 3px;' alt='{service['name']} logo'>\n"
                        output += f"<span style='font-size: 0.8em; font-weight: 500; color: #2c3e50;'>{service['name']}</span>\n"
                        output += f"</div>\n"
                    else:
                        output += f"<div style='background: white; padding: 4px 8px; border-radius: 6px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); border: 1px solid #ddd; font-size: 0.8em; font-weight: 500; color: #2c3e50;'>{service['name']}</div>\n"
                output += f"</div>\n"
                output += f"</div>\n"
            
            # Subscription services
            subscription_services = movie.get('subscription_services', [])
            if subscription_services:
                output += f"<div>\n"
                output += f"<div style='font-weight: bold; color: #2c5530; margin-bottom: 6px; font-size: 0.9em; text-align: center;'>💳 Free with Subscription</div>\n"
                output += f"<div style='display: flex; flex-wrap: wrap; gap: 6px; justify-content: center;'>\n"
                for service in subscription_services:
                    if service.get('logo_url'):
                        output += f"<div style='display: flex; align-items: center; background: white; padding: 4px 8px; border-radius: 6px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); border: 1px solid #ddd;'>\n"
                        output += f"<img src='{service['logo_url']}' style='width: 20px; height: 20px; margin-right: 4px; border-radius: 3px;' alt='{service['name']} logo'>\n"
                        output += f"<span style='font-size: 0.8em; font-weight: 500; color: #2c3e50;'>{service['name']}</span>\n"
                        output += f"</div>\n"
                    else:
                        output += f"<div style='background: white; padding: 4px 8px; border-radius: 6px; box-shadow: 0 2px 4px rgba(0,0,0,0.1); border: 1px solid #ddd; font-size: 0.8em; font-weight: 500; color: #2c3e50;'>{service['name']}</div>\n"
                output += f"</div>\n"
                output += f"</div>\n"
            
            # JustWatch attribution (required by TMDB terms)
            output += f"<div style='margin-top: 8px; font-size: 0.7em; color: #666; text-align: center;'>Streaming data provided by <strong>JustWatch</strong></div>\n"
            output += f"</div>\n"
        
        # Box Office (if available) - more compact
        if movie.get('has_omdb_data') and movie.get('box_office') != 'N/A':
            output += f"<p style='margin: 8px 0; color: #27ae60; background: #eafaf1; padding: 8px; border-radius: 6px; font-size: 0.9em; text-align: center;'><strong>💰 Box Office:</strong> {movie['box_office']}</p>\n"
        
        # External Links - centered and more compact
        links_html = ""
        if movie.get('imdb_url'):
            links_html += f"<div style='text-align: center; margin: 12px 0;'>\n"
            links_html += f"<a href='{movie['imdb_url']}' target='_blank' style='display: inline-block; background: linear-gradient(45deg, #f5c518, #f39c12); color: #000; padding: 8px 16px; text-decoration: none; border-radius: 6px; font-weight: bold; font-size: 0.9em; box-shadow: 0 3px 6px rgba(0,0,0,0.2); transition: transform 0.2s;' onmouseover='this.style.transform=\"translateY(-1px)\"' onmouseout='this.style.transform=\"translateY(0)\"'>🎬 View on IMDb</a>\n"
            links_html += f"</div>\n"
        
        if links_html:
            output += links_html

        output += f"</div>\n\n"
    
    return output

# Create simple Gradio interface
with gr.Blocks(
    title="Movie Recommendation", 
    theme="gstaff/xkcd",
    css="""
        footer {
            display: none !important;
        }
        .gradio-container .footer {
            display: none !important;
        }
        .gradio-container .footer-text {
            display: none !important;
        }
        .gradio-container .built-with {
            display: none !important;
        }
    """
) as app:
    gr.Markdown("# 🎬 Random Movie Recommendations")
    gr.Markdown("Select what you are in the mood for and get a movie recommendation!")
    
    with gr.Row():
        with gr.Column():
            decade = gr.Dropdown(
                choices=config.DECADES,
                label="Decade",
                value="2000s"
            )
            
            country = gr.Dropdown(
                choices=config.COUNTRIES,
                label="Country",
                value="United States of America (US)",
                filterable=True
            )
            
            genre = gr.Dropdown(
                choices=config.GENRES,
                label="Genre",
                value="Action",
                visible=True  # Initially visible since default is US
            )
            
            num_movies = gr.Slider(
                minimum=1,
                maximum=10,
                value=5,
                step=1,
                label="Number of Movies"
            )
            
            get_btn = gr.Button("Get Recommendations", variant="primary")
    
    results = gr.Markdown(label="Recommendations")
    
    # Function to update genre visibility based on country selection
    def update_genre_visibility(selected_country):
        if selected_country in config.TOP_COUNTRIES:
            return gr.update(visible=True)
        else:
            return gr.update(visible=False, value=None)
    
    # Update genre dropdown when country changes
    country.change(
        update_genre_visibility,
        inputs=[country],
        outputs=[genre]
    )
    
    # Connect the recommendation function
    get_btn.click(
        get_movie_recommendations,
        inputs=[decade, country, genre, num_movies],
        outputs=results
    )

if __name__ == "__main__":
    # Validate configuration before starting the app
    try:
        config.validate()
        logger.info("Starting Movie Recommendation App")
        app.launch()
    except Exception as e:
        logger.error(f"Failed to start application: {e}")
        raise