update

app.py

@@ -22,23 +22,23 @@ async def generate_context(summaries, system_prompt_prefix, system_prompt_suffix
         payload = {
             "model": "gpt-4-turbo-preview",
             "messages": [
-                        {"role": "system", "content": """
-System prompt:システムプロンプトは、コンテンツの前提条件を作るために使用されます。
-You are an AI assistant. Based on the document summary below, please provide a concise context that captures the essential information:
-ご自由にお代えください。
-                         """ 
-                         },
+                        {"role": "system", "content": "You are an AI assistant. Based on the document summary below, please provide a concise context that captures the essential information:"},
                         {"role": "user", "content": prompt}
                         ],
             "temperature": 0.7,
             "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()
+        try:
+            async with session.post(API_URL, json=payload, headers=headers) as response:
+                if response.status != 200:
+                    raise Exception(f"Error generating context: {response.status} - {await response.text()}")
+                result = await response.json()
+                return result["choices"][0]["message"]["content"].strip()
+        except Exception as e:
+            st.error(f"Error generating context: {str(e)}")
+            return None
 
-# 2nd Step: Generate Scripts
 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}"
 
@@ -46,32 +46,35 @@ async def generate_script(context, heading, system_prompt_prefix, system_prompt_
         payload = {
             "model": "gpt-4-turbo-preview",
             "messages": [
-                        {"role": "system", "content": """
-
-
-
-System prompt:システムプロンプトは、コンテンツの前提条件を作るために使用されます。"""
-                        },
+                        {"role": "system", "content": f"You are an AI scriptwriter. Your task is to generate a detailed script based on the given context and the provided heading."},
                         {"role": "user", "content": prompt}
                         ],
             "temperature": 0.7,
             "max_tokens": 4000,
         }
         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()
+        try:
+            async with session.post(API_URL, json=payload, headers=headers) as response:
+                if response.status != 200:
+                    raise Exception(f"Error generating script: {response.status} - {await response.text()}")
+                result = await response.json()
+                return result["choices"][0]["message"]["content"].strip()
+        except Exception as e:
+            st.error(f"Error generating script: {str(e)}")
+            return None
+
+
 
 async def main():
     st.title("Script Generator")
 
-    context_system_prompt_prefix = st.text_input("Enter the prompt prefix for context generation:", value="")
+    context_system_prompt_prefix = st.text_input("Enter the prompt prefix for context generation:", value="You are an AI assistant. Your task is to provide a concise context that captures the essential information from the given document summary.")
     summaries = st.text_area("Enter the document summary:", height=200)
-    context_system_prompt_suffix = st.text_input("Enter the prompt suffix for context generation:", value="")
+    context_system_prompt_suffix = st.text_input("Enter the prompt suffix for context generation:", value="Focus on the key points and main ideas, and present the context in a clear and organized manner.")
 
     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="")
+    script_system_prompt_prefix = st.text_input("Enter the system prompt prefix for script generation:", value="You are an AI scriptwriter. Your task is to generate a detailed script based on the given context and the provided heading.")
+    script_system_prompt_suffix = st.text_input("Enter the system prompt suffix for script generation:", value="The script should be relevant, informative, engaging, and contain more than 600 characters.")
 
     if st.button("Generate Context and Scripts"):
         with st.spinner("Generating context..."):
@@ -87,7 +90,4 @@ async def main():
         st.header("Generated Scripts")
         for heading, script in zip(headings, scripts):
             st.subheader(heading)
-            st.write(script)
-
-if __name__ == "__main__":
-    asyncio.run(main())
+            st.write(script)