Create App.py

aap.py

@@ -0,0 +1,67 @@
+import os
+import asyncio
+import streamlit as st
+import aiohttp
+
+API_KEY = os.environ["OPENAI_API_KEY"]
+API_URL = "https://api.openai.com/v1/chat/completions"
+
+async def generate_context(summaries, system_prompt_prefix, system_prompt_suffix):
+    prompt = f"{system_prompt_prefix}\n\nYou are an AI assistant. Based on the document summary below, please provide a concise context that captures the essential information:\n\n{summaries}\n\n{system_prompt_suffix}"
+
+    async with aiohttp.ClientSession() as session:
+        payload = {
+            "model": "gpt-4-turbo-preview",
+            "messages": [{"role": "user", "content": prompt}],
+            "temperature": 0,
+            "max_tokens": 2048,
+        }
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        async with session.post(API_URL, json=payload, headers=headers) as response:
+            result = await response.json()
+            return result["choices"][0]["message"]["content"].strip()
+
+async def generate_script(context, heading, system_prompt_prefix, system_prompt_suffix):
+    prompt = f"{system_prompt_prefix}\n\n{context}\n\nYou are an AI scriptwriter. Based on the context above and the heading provided below, please generate a detailed script of more than 600 characters.\n\nHeading: {heading}\n\n{system_prompt_suffix}"
+
+    async with aiohttp.ClientSession() as session:
+        payload = {
+            "model": "gpt-4-turbo-preview",
+            "messages": [{"role": "user", "content": prompt}],
+            "temperature": 0,
+            "max_tokens": 2048,
+        }
+        headers = {"Authorization": f"Bearer {API_KEY}"}
+        async with session.post(API_URL, json=payload, headers=headers) as response:
+            result = await response.json()
+            return result["choices"][0]["message"]["content"].strip()
+
+async def main():
+    st.title("Script Generator")
+
+    summaries = st.text_area("Enter the document summary:", height=200)
+    context_system_prompt_prefix = st.text_input("Enter the system prompt prefix for context generation:", value="")
+    context_system_prompt_suffix = st.text_input("Enter the system prompt suffix for context generation:", value="")
+
+    outline = st.text_area("Enter the content outline (one heading per line):", height=200)
+    script_system_prompt_prefix = st.text_input("Enter the system prompt prefix for script generation:", value="")
+    script_system_prompt_suffix = st.text_input("Enter the system prompt suffix for script generation:", value="")
+
+    if st.button("Generate Context and Scripts"):
+        with st.spinner("Generating context..."):
+            context = await generate_context(summaries, context_system_prompt_prefix, context_system_prompt_suffix)
+        st.header("Generated Context")
+        st.write(context)
+
+        headings = outline.split("\n")
+        with st.spinner("Generating scripts..."):
+            tasks = [generate_script(context, heading, script_system_prompt_prefix, script_system_prompt_suffix) for heading in headings]
+            scripts = await asyncio.gather(*tasks)
+
+        st.header("Generated Scripts")
+        for heading, script in zip(headings, scripts):
+            st.subheader(heading)
+            st.write(script)
+
+if __name__ == "__main__":
+    asyncio.run(main())