import requests
import streamlit as st

API_URL = "http://127.0.0.1:5000"  # Replace with actual backend URL

st.title("Text & Voice to Image Generator")

# Text input
user_input = st.text_area("Enter your text:")
if st.button("Generate from Text"):
    response = requests.post(f"{API_URL}/process_text", json={"text": user_input}).json()
    st.write("Generated Response:", response["response"])
    st.image(response["image"], caption="Generated Image")

# Audio input
st.write("Upload an audio file:")
audio_file = st.file_uploader("Upload your voice (MP3/WAV)", type=["mp3", "wav"])
if audio_file and st.button("Generate from Audio"):
    files = {"file": audio_file.getvalue()}
    response = requests.post(f"{API_URL}/process_audio", files=files).json()
    st.write("Transcribed Text:", response["transcribed_text"])
    st.write("Generated Response:", response["response"])
    st.image(response["image"], caption="Generated Image")