components/display_variations.py

import pandas as pd
import streamlit as st

def display_script_variations(json_data: dict):
    if not json_data or "script_variations" not in json_data:
        st.error("No script variations found")
        return
    for i, variation in enumerate(json_data["script_variations"], 1):
        st.markdown(f"### Variation {i}: {variation.get('variation_name','Var')}")
        df = pd.DataFrame(variation.get("script_table", []))
        st.table(df)
    csv_content = pd.concat(
        [pd.DataFrame(v.get("script_table", []))
         .assign(Variation=v.get("variation_name", f"Var{i+1}"))
         for i, v in enumerate(json_data["script_variations"])],
        ignore_index=True
    ).to_csv(index=False)
    st.download_button("Download CSV", data=csv_content,
                       file_name="scripts.csv", mime="text/csv")