update

app.py

@@ -1,3 +1,6 @@
+from io import BytesIO
+
+import pandas as pd
 import requests
 import streamlit as st
 import json
@@ -104,10 +107,26 @@ values = st.text_area("Company Values", "innovation, teamwork, transparency")
 batch_size = st.number_input("Batch Size", min_value=1, max_value=100, value=10)
 repeat_times = st.number_input("Number of Batches", min_value=1, max_value=10, value=1)
 
+def to_excel(df):
+    output = BytesIO()
+    writer = pd.ExcelWriter(output, engine='xlsxwriter')
+    df.to_excel(writer, index=False, sheet_name='Potshots')
+    writer.save()
+    processed_data = output.getvalue()
+    return processed_data
 
 
 if st.button("Generate Potshots"):
     with st.spinner("Generating customized potshots..."):
-        results = get_potshots(n_repeat=repeat_times, batch=batch_size, role=role, tone=tone, audience=audience,
-                               values=values)
-        st.json(results)
+        df = get_potshots(n_repeat=repeat_times, batch=batch_size, role=role, tone=tone, audience=audience,
+                          values=values)
+
+        # Display the generated potshots as a table
+        st.write("### Generated Potshots")
+        st.dataframe(df)
+
+        # Provide a download button for the Excel file
+        st.write("### Download Potshots")
+        excel_data = to_excel(df)
+        st.download_button(label="Download Potshots as Excel", data=excel_data, file_name="potshots.xlsx",
+                           mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet")

requirements.txt

@@ -1,3 +1,4 @@
 openai
 json-repair
-backoff
+backoff
+pandas