import streamlit as st
import requests

# --- Securely store your API keys here ---
OPENROUTER_API_KEY = "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"  # Replace with your actual key
YOUTUBE_API_KEY = "AIzaSyCFE-fnv1hKm6VW7xN6RNGPSaSG4qKam94"  # Replace with your actual YouTube Data API key

# --- Function to interact with OpenRouter API ---
def get_ai_solution(problem):
    headers = {
        'Authorization': f'Bearer {OPENROUTER_API_KEY}',
        'Content-Type': 'application/json',
    }
    data = {
        'model': 'mistralai/mistral-7b-instruct',
        'messages': [
            {
                "role": "user",
                "content": f"What are the common repair suggestions for this problem: '{problem}'"
            }
        ]
    }

    response = requests.post("https://openrouter.ai/api/v1/chat/completions", headers=headers, json=data)

    if response.status_code == 200:
        result = response.json()
        return result['choices'][0]['message']['content']
    else:
        return f"Error: {response.text}"

# --- Function to find YouTube video relevant to the problem ---
def find_youtube_video(problem):
    search_url = "https://www.googleapis.com/youtube/v3/search"
    params = {
        "part": "snippet",
        "q": problem,
        "key": YOUTUBE_API_KEY,
        "type": "video",
        "maxResults": 1
    }

    response = requests.get(search_url, params=params)
    if response.status_code == 200:
        data = response.json()
        if data["items"]:
            video_id = data["items"][0]["id"]["videoId"]
            return f"https://www.youtube.com/watch?v={video_id}"
    return None

# --- Streamlit App UI ---
st.set_page_config(page_title="🔧 AI Mechanic Assistant", layout="centered")
st.title("🔧 AI Mechanic Assistant")
st.caption("Get AI-powered solutions and video tutorials for vehicle problems.")

problem = st.text_area("Describe your vehicle problem:")

if st.button("Get Solution"):
    if problem.strip():
        with st.spinner("Thinking..."):
            solution = get_ai_solution(problem)
            video_url = find_youtube_video(problem)

        st.subheader("🧠 Suggested Fix:")
        st.write(solution)

        if video_url:
            st.subheader("📺 Related Video:")
            st.video(video_url)
        else:
            st.info("No related YouTube video found.")
    else:
        st.warning("Please describe the problem first.")