src/streamlit_app.py

import streamlit as st

from authen.authentication import login_gate, logout
from app_pages.image_generation import render_image_generation_page
from app_pages.Image_gen_library import render_image_gen_library

st.set_page_config(page_title="ImageAdGen", layout="wide")


uid = login_gate()
st.session_state["uid"] = uid

st.title("Ad Prompt + Image Generator")


with st.sidebar:
    st.markdown(f"**Signed in as:** `{uid}`")
    if st.button("Log out", key="btn_logout"):
        logout()
        st.rerun()

tabs = st.tabs(["Generate", "Library"])

with tabs[0]:
    render_image_generation_page(uid=uid)

with tabs[1]:
    render_image_gen_library(uid=uid)