app.py

from dotenv import load_dotenv
import streamlit as st
import os
import google.generativeai as genai
from PIL import Image

# Configure page
st.set_page_config(page_title="OtakuOracle 🌟", layout="wide")

# Custom CSS for styling
st.markdown("""
<style>
    .stApp {
        background: linear-gradient(135deg, #1a1a2e 0%, #16213e 100%);
        color: #ffffff;
    }
    .stButton button {
        background-color: #ff6b6b;
        color: white;
        border-radius: 10px;
        transition: all 0.3s ease;
    }
    .stButton button:hover {
        background-color: #ff8787;
        transform: translateY(-2px);
        box-shadow: 0 5px 15px rgba(255, 107, 107, 0.4);
    }
    .sidebar .sidebar-content {
        background: rgba(26, 26, 46, 0.9);
    }
    .stTextInput input {
        border-radius: 8px;
        border: 2px solid #4a90e2;
    }
    .css-1d391kg {
        background-color: rgba(255, 255, 255, 0.05);
        border-radius: 15px;
        padding: 20px;
        margin: 10px 0;
    }
</style>
""", unsafe_allow_html=True)

# API Key input in sidebar
api_key = st.sidebar.text_input("Enter Google API Key 🔑", type="password")
if not api_key:
    st.warning("⚠️ Please enter your Google API Key to use the app")
    st.stop()

# Configure Gemini with provided API key
genai.configure(api_key=api_key)

def get_anime_recommendations(anime_title):
    query = f"Give me 5 anime recommendations similar to {anime_title} with the following details: Anime, Genre, Main Protagonists, Summary."
    response = chat.send_message(query, stream=True)
    full_response = ""
    for chunk in response:
        full_response += chunk.text
    return full_response

def format_anime_recommendation(text):
    recommendations = text.split('\n\n')
    formatted_recommendations = []
    for rec in recommendations:
        lines = rec.split('\n')
        if len(lines) >= 4:
            anime = lines[0].replace("Anime:", "").strip()
            genre = lines[1].replace("Genre:", "").strip()
            protagonists = lines[2].replace("Main Protagonists:", "").strip()
            summary = lines[3].replace("Summary:", "").strip()
            formatted_recommendations.append({
                "Anime": anime,
                "Genre": genre,
                "Main Protagonists": protagonists,
                "Summary": summary
            })
    return formatted_recommendations

def get_anime_character_response(character, question):
    query = f"Respond to the following question in the style of the anime character {character}: {question}"
    response = chat.send_message(query, stream=True)
    full_response = ""
    for chunk in response:
        full_response += chunk.text
    return full_response

def get_gemini_response(input, image, prompt):
    model = genai.GenerativeModel('gemini-1.5-flash')
    response = model.generate_content([input, image[0], prompt])
    return response.text

def input_image_setup(uploaded_file):
    if uploaded_file is not None:
        bytes_data = uploaded_file.getvalue()
        image_parts = [
            {
                "mime_type": uploaded_file.type,
                "data": bytes_data
            }
        ]
        return image_parts
    else:
        raise FileNotFoundError("No file uploaded")

def load_animation():
    with st.spinner('🌟 Processing...'):
        st.balloons()

# Initialize the Gemini Pro model
chat = genai.GenerativeModel("gemini-1.5-flash-002")
chat = chat.start_chat(history=[])

# Sidebar navigation
st.sidebar.markdown("## Navigation Menu 🎮")
page = st.sidebar.selectbox("Select Service", 
    ["Anime Recommender 🌟", "Talk to Anime Character 🗣️", "Anime Face Detect 👀"])

# Anime Recommender Page
if page == "Anime Recommender 🌟":
    st.markdown("# 🎬 Anime Recommender System")
    st.markdown("---")
    
    input_anime = st.text_input("Input Anime Title: ", 
                               key="input_anime", 
                               help="Enter the title of an anime you like!")
    
    if st.button("Get Recommendations 🚀"):
        if input_anime:
            load_animation()
            response = get_anime_recommendations(input_anime)
            recommendations = format_anime_recommendation(response)
            
            st.subheader("Anime Recommendations 📜")
            for rec in recommendations:
                with st.container():
                    st.markdown(f"""
                    **Anime:** {rec['Anime']} 📺  
                    **Genre:** {rec['Genre']} 🎭  
                    **Main Protagonists:** {rec['Main Protagonists']} 🧑‍🤝‍🧑  
                    **Summary:** {rec['Summary']} 📝
                    ---
                    """)

# Anime Character Chat Page
elif page == "Talk to Anime Character 🗣️":
    st.markdown("# 💬 Chat with an Anime Character")
    st.markdown("---")
    
    character = st.text_input("Anime Character Name:", 
                             help="Enter character name!")
    question = st.text_input("Your Question:", 
                            help="Ask your question!")
    
    if st.button("Ask Character 🗨️"):
        if character and question:
            load_animation()
            response = get_anime_character_response(character, question)
            st.markdown(f"### Response from {character} 🎭")
            st.write(response)

# Anime Face Detect Page
elif page == "Anime Face Detect 👀":
    st.markdown("# 🕵️ Anime Character Detector")
    st.markdown("---")

    input_prompt = """
    You are an expert in the world of Anime where you need to see characters from the image and provide the details of every character in this below format:

    1. Name of the Character(s):
    2. Anime:
    3. Role:
    4. Favourite Habit:

    If there are multiple characters then mention them separately in the above format

    Note: If you don't find the character related to any Anime then kindly tell the user to provide an Anime character picture in this below format:
    'Kindly provide a character from Anime'
    """

    input_text = st.text_input("Input Prompt:", 
                              help="Enter custom prompt!")
    uploaded_file = st.file_uploader("Choose an image... 📷", 
                                    type=["jpg", "jpeg", "png"])
    
    if uploaded_file:
        image = Image.open(uploaded_file)
        st.image(image, caption="Uploaded Image 🖼️", use_column_width=True)

        if st.button("Analyze Character 🔍"):
            load_animation()
            image_data = input_image_setup(uploaded_file)
            response = get_gemini_response(input_text, image_data, input_prompt)
            st.markdown("### Analysis Results 📝")
            st.write(response)