streamlit_ui - Copy.py

# streamlit_ui.py

import streamlit as st
import requests
import base64
from PIL import Image
import io
import os
import tempfile

st.set_page_config(
    page_title="Prompta - Text to Media Generator",
    page_icon="🎙️",
    layout="wide",
    initial_sidebar_state="expanded"
)
st.title("🎙️🖼️🎞️ Prompta - Text to Media Generator")

# 🛠️ Get Token FIRST
TOKEN = st.sidebar.text_input("🔑 API Token", type="password")
HEADERS = {"Authorization": f"Bearer {TOKEN}"} if TOKEN else {}

# ✅ Display AFTER token is typed
if TOKEN:
    st.sidebar.write("Using token:", TOKEN)
    st.sidebar.write("Sending headers:", HEADERS)
else:
    st.sidebar.warning("⚠️ Please enter a valid API token to use the app.")

API_BASE = "http://localhost:8000"

#API_BASE = "https://2255d6a4793d.ngrok-free.app"

def render_media(file_bytes, media_type, caption):
    b64 = base64.b64encode(file_bytes).decode()
    if media_type == "audio":
        st.audio(f"data:audio/wav;base64,{b64}", format="audio/wav")
    elif media_type == "video":
        st.video(f"data:video/mp4;base64,{b64}")
    elif media_type == "image":
        try:
            # Validate if it's a valid image
            img = Image.open(io.BytesIO(file_bytes))
            st.image(img, caption=caption)
        except Exception as e:
            st.warning("⚠️ Cannot render image. It may be corrupt or empty.")
            st.code(str(e))

# Sidebar inputs
st.sidebar.header("🛠️ Settings")
#TOKEN = st.sidebar.text_input("🔑 API Token", type="password")
#HEADERS = {"Authorization": f"Bearer {TOKEN}"} if TOKEN else {}

voice = st.selectbox("Choose voice", ["en", "hi", "te", "ta"])
voice_map = {
    "en": "en-US",
    "hi": "hi-IN",
    "te": "te-IN",
    "ta": "ta-IN"
}

tab = st.sidebar.radio("Select Task", ["Text to Audio", "Text to Image", "Text to Video"])

if tab == "Text to Audio":
    st.subheader("🎤 Text to Audio")
    text = st.text_area("Enter text")
    voice = st.selectbox("Choose language", ["English", "Hindi", "Telugu", "Tamil"])
    voice_map = {
        "English": ("en-US", "en"),
        "Hindi": ("hi-IN", "hi"),
        "Telugu": ("te-IN", "te"),
        "Tamil": ("ta-IN", "ta")
    }
    voice_code, lang_code = voice_map[voice]

    if st.button("🔊 Generate Audio"):
        with st.spinner("Generating audio..."):
            r = requests.post(
                f"{API_BASE}/api/v1/audio/generate", 
                json={
                    "text": text,
                    "voice": voice_code,
                    "language": lang_code
                },
                headers=HEADERS
            )
            if r.status_code == 200:
                try:
                    data = r.json()
                    st.code(data, language="json")  # Debug: show full JSON response in UI

                    if "download_url" in data:
                        download_url = f"{API_BASE}{data['download_url']}"
                        audio_resp = requests.get(download_url, headers=HEADERS)
                        if audio_resp.status_code == 200:
                            render_media(audio_resp.content, "audio", "Generated Audio")
                        else:
                            st.error("❌ Failed to download audio file.")
                    else:
                        st.error("❌ `download_url` not found in API response.")
                        st.code(data)
                except Exception as e:
                    st.error("❌ Failed to parse API response.")
                    st.code(r.text)
                    st.exception(e)
            else:
                st.error(f"❌ Failed: {r.json().get('detail')}")


elif tab == "Text to Image":
    st.subheader("🖼️ Text to Image")
    prompt = st.text_area("Enter image prompt")
    style = st.selectbox("Choose Style", ["sdxl", "deepfloyd", "kandinsky"])

    if st.button("🧠 Generate Image"):
        with st.spinner("Generating image..."):
            r = requests.post(
                f"{API_BASE}/api/v1/image/generate", 
                json={"prompt": prompt, "style": style},  # ✅ correct key
                headers=HEADERS
            )
            if r.status_code == 200:
                try:
                    res_json = r.json()
                    download_url = res_json.get("download_url")
                    if not download_url:
                        st.error("No download URL returned.")
                    else:
                        download_full_url = f"{API_BASE}{download_url}"
                        image_response = requests.get(download_full_url, headers={"accept": "image/png"}, allow_redirects=True)
                        if image_response.status_code != 200:
                            st.error("❌ Failed to download image.")
                            st.code(image_response.text)
                            st.write("Status:", image_response.status_code)
                            st.write("Headers:", image_response.headers)
                        st.write(image_response.status_code, image_response.headers)
                        render_media(image_response.content, "image", "Generated Image")
                except Exception as e:
                    st.error(f"⚠️ Failed to fetch/display image: {str(e)}")
                    st.code(r.text)
            else:
                try:
                    detail = r.json().get("detail")
                except Exception:
                    detail = r.text  # fallback to raw response text (may be empty or HTML)

                st.error(f"❌ Failed: {detail}")                

elif tab == "Text to Video":
    st.subheader("🎞️ Text to Video")
    prompt = st.text_area("Enter video prompt")
    tone = st.selectbox("Tone", ["formal", "casual", "emotional", "documentary"])
    domain = st.selectbox("Domain", ["health", "education", "governance", "entertainment"])
    environment = st.selectbox("Environment", ["urban", "rural", "nature", "futuristic"])

    transcript = st.text_area("Transcript (optional - for subtitles)", height=100)
    enhance = st.checkbox("✨ Add Subtitles and Background Music")

    if st.button("🎬 Generate Video"):
        with st.spinner("Generating video..."):
            r = requests.post(
                f"{API_BASE}/api/v1/video/generate",
                json={"prompt": prompt, "tone": tone, "domain": domain, "environment": environment},
                headers=HEADERS
            )
            if r.status_code == 200:
                try:
                    data = r.json()
                    st.code(data, language="json")

                    download_url = data.get("download_url")
                    if not download_url:
                        st.error("⚠️ No download URL received.")
                    else:
                        full_video_url = f"{API_BASE}{download_url}"
                        video_response = requests.get(full_video_url, headers=HEADERS)
                        if video_response.status_code == 200:
                            video_bytes = video_response.content
                            st.write("📦 Video size (bytes):", len(video_bytes))

                            if enhance and transcript:
                                with tempfile.NamedTemporaryFile(delete=False, suffix=".mp4") as tmp_vid:
                                    tmp_vid.write(video_bytes)
                                    tmp_vid_path = tmp_vid.name

                                srt_path = generate_srt_from_text(transcript, output_path="streamlit_subs.srt")
                                enhanced_path = "streamlit_final_video.mp4"
                                enhance_video_with_subtitles_and_bgm(
                                    video_path=tmp_vid_path,
                                    srt_path=srt_path,
                                    bgm_path="default_bgm.mp3",
                                    output_path=enhanced_path
                                )

                                with open(enhanced_path, "rb") as f:
                                    render_media(f.read(), "video", "Enhanced Video")
                            else:
                                st.video(video_bytes)
                        else:
                            st.error("❌ Failed to download video.")
                except Exception as e:
                    st.error("❌ Error parsing response or rendering video.")
                    st.code(r.text)
                    st.exception(e)
            else:
                try:
                    st.error(f"❌ Failed: {r.json().get('detail')}")
                except:
                    st.error(f"❌ Failed: {r.text}")


st.sidebar.markdown("---")
st.sidebar.info("Built with ❤️ for AI GovTech Challenge 2025")