Update app.py

app.py

@@ -1,24 +1,31 @@
 import gradio as gr
 import openai
 
-def generate_scene_description(api_key, theme, scene_number):
+def generate_scene_descriptions(api_key, theme):
     openai.api_key = api_key
     
-    # Variables that represent each part of the scene description
-    intro = f"Generate a structured scene description for a video on the theme \"{theme}\":\n\n"
-    scene = f"Scene #{scene_number}\n"
-    narrator_voice = "Narrator's voice: \n"
-    video_prompt = "Video prompt: \n"
-    voice_over = "Voice over: \n"
-    music_style = "Music style:  \n"
+    # Base introduction for the prompt
+    intro = f"Generate structured scene descriptions for a series of scenes for a video on the theme \"{theme}\":\n"
     
-    prompt = intro + scene + narrator_voice + video_prompt + voice_over + music_style
+    # Constructing the prompt for multiple scenes
+    scenes_prompt = ""
+    for i in range(1, 6):  # For generating up to 5 scenes
+        scenes_prompt += (
+            f"Scene #{i}\n"
+            "Narrator's voice: \n"
+            "Video prompt: \n"
+            "Voice over: \n"
+            "Music style:  \n\n"
+        )
+
+    # Combine intro and scenes into the full prompt
+    full_prompt = intro + scenes_prompt
     
     # Call to OpenAI's GPT-3.5 API with the custom prompt
     response = openai.Completion.create(
         engine="text-davinci-003",
-        prompt=prompt,
-        max_tokens=550
+        prompt=full_prompt,
+        max_tokens=550  # Adjust as necessary
     )
     
     # The response from the AI is formatted text based on our structured prompt
@@ -26,16 +33,14 @@ def generate_scene_description(api_key, theme, scene_number):
 
 # Create the Gradio interface with appropriate inputs and outputs
 iface = gr.Interface(
-    fn=generate_scene_description,
+    fn=generate_scene_descriptions,
     inputs=[
-        gr.Textbox(label="Thematic Area", placeholder="Enter a thematic area, e.g., Climate Crisis"),  # Thematic area input
         gr.Textbox(label="Your OpenAI API Key", type="password"),  # Secure API key input
-
-        #        gr.Dropdown(label="Scene Number", choices=[1, 2, 3, 4, 5, 6], type="index")  # Dropdown for selecting scene number
+        gr.Textbox(label="Thematic Area", placeholder="Enter a thematic area, e.g., Climate Crisis")  # Thematic area input
     ],
     outputs="text",  # The output is a text field
-    title="Scene Generator",  # Title of the Gradio app
-    description="This app generates structured scene descriptions for a video based on the given thematic area using GPT-3.5."  # Description for the Gradio app
+    title="Multi-Scene Generator",  # Title of the Gradio app
+    description="This app generates structured scene descriptions for multiple scenes of a video based on the given thematic area using GPT-3.5."  # Description for the Gradio app
 )
 
 # Launch the interface locally for testing