import streamlit as st
import requests

# Streamlit interface setup
st.title('Video Summary Interface')

# Input for modifying the prompt
prompt = st.text_input("Enter your prompt:", 
                       "list the top 4 job interview mistakes and how to improve")

# Slider to adjust the number in the prompt
number = st.slider("Select the number of top mistakes:", min_value=1, max_value=10, value=4)

# Update the prompt with the chosen number
updated_prompt = prompt.replace("4", str(number))

# Button to send the request
if st.button("Summarize Video"):
    BASE_URL = "https://api.twelvelabs.io/v1.2"
    api_key = "tlk_3CPMVGM0ZPTKNT2TKQ3Y62TA7ZY9"
    data = {
        "video_id": "6636cf7fd1cd5a287c957cf5",
        "type": "summary",
        "prompt": updated_prompt
    }

    # Send the request
    response = requests.post(f"{BASE_URL}/summarize", json=data, headers={"x-api-key": api_key})
    
    # Check if the response is successful
    if response.status_code == 200:
        st.text_area("Summary:", response.json()['summary'], height=300)
    else:
        st.error("Failed to fetch summary: " + response.text)

# Run this script using the following command:
# streamlit run your_script_name.py