import os
import mimetypes
import cv2
import uuid
import streamlit as st
from google import generativeai as genai
# import pywhatkit 
import urllib.parse

# Configure Gemini API
def configure_api():
    api_key = "AIzaSyBDkLcFKzE_T6r1E5XjbuXVaoW40Szn71s"  # Use your real key here
    genai.configure(api_key=api_key)
    return genai.GenerativeModel('gemini-2.0-flash')

def capture_from_webcam():
    cam = cv2.VideoCapture(0)
    if not cam.isOpened():
        st.error("Could not open webcam.")
        return None

    st.info("Capturing image... Press SPACE to capture.")
    filename = "ahmad.jpg"
    while True:
        ret, frame = cam.read()
        if not ret:
            st.error("Failed to grab frame.")
            break

        cv2.imshow("Camera Feed - Press SPACE to capture", frame)
        key = cv2.waitKey(1)
        if key % 256 == 32:  # SPACE pressed
            cv2.imwrite(filename, frame)
            break
        elif key % 256 == 27:  # ESC
            break

    cam.release()
    cv2.destroyAllWindows()
    if os.path.exists(filename):
        return filename
    return None

def process_image(image_path, model):
    mime_type, _ = mimetypes.guess_type(image_path)
    if not mime_type:
        mime_type = "image/jpeg"

    with open(image_path, "rb") as img_file:
        image_data = img_file.read()

    response = model.generate_content(
        [
            {"mime_type": mime_type, "data": image_data},
            """complete analyze the given picture pay attention any detail of picture and tell which pakistani song is suitble according person in picture 
            just give me the name of one hindi song according to peroson in pic and just give the name of song no other detail just name ."""
        ]
    )
    return response.text.strip()

# Streamlit UI
st.set_page_config(page_title="Hindi Song Recommender", layout="centered")
st.title("🎶 Hindi Song Recommender from Image")

st.write("### Aap image kaise doge?")
option = st.radio("Choose input method:", ["Upload Image", "Capture from Webcam"])

model = configure_api()

if option == "Upload Image":
    uploaded_file = st.file_uploader("Drag and drop your image here", type=["jpg", "jpeg", "png"])
    if uploaded_file:
        path = f"temp_{uuid.uuid4().hex[:8]}.jpg"
        with open(path, "wb") as f:
            f.write(uploaded_file.read())
        st.session_state.image_path = path
        st.image(path, caption="Uploaded Image", use_container_width=True)

elif option == "Capture from Webcam":
    st.write("📸 Use your device's camera:")
    camera_photo = st.camera_input("Take a picture")
    
    if camera_photo is not None:
        # Save the captured image
        path = f"captured_{uuid.uuid4().hex[:8]}.jpg"
        with open(path, "wb") as f:
            f.write(camera_photo.read())
        st.session_state.image_path = path
        st.image(path, caption="Captured Image", use_container_width=True)


if "image_path" in st.session_state and st.button("Suggest Hindi Song"):
    with st.spinner("Analyzing image and finding a song..."):
        try:
            suggested_song = process_image(st.session_state.image_path, model)
            st.success(f"🎵 Suggested Hindi Song: {suggested_song}")
            search_query = urllib.parse.quote_plus(suggested_song + " song")
            youtube_link = f"https://www.youtube.com/results?search_query={search_query}"
            st.markdown(f"[🔗 Click to search on YouTube]({youtube_link})", unsafe_allow_html=True)
        except Exception as e:
            st.error(f"Error: {e}")