Update app.py

app.py

@@ -156,6 +156,8 @@ def prompt_to_video(video_prompt):
         The paragraph in the slides should be formatted in bullet points and each bullet point should be 1.5 line spacing apart.
         Header and paragraph should be aligned in an aesthetically pleasing way.
         Return the output as a nicely formatted HTML string.
+        Header should be aligned to the center.
+        Font color should be white and background black.
         Font should be Roboto.
         Do not say "Here's your" or "Sure".  Directly output the HTML string only.
         Make sure there is nothing before or after the HTML string.  ONLY output the HTML string.
@@ -163,23 +165,24 @@ def prompt_to_video(video_prompt):
         Do not declare a new variable, output the HTML string only.
         '''
         html_out_list.append(chatgpt_chain.predict(human_input=prompt_input2))
-    
-    os.makedirs("slide", exist_ok=True)
+
+    extract_path = 'slide_' + video_prompt
+    os.makedirs(extract_path, exist_ok=True)
         
     num = 1
     for html_string in html_out_list:
         print(html_string)
-        with open(f"slide/slide_{num}.html", "w") as file:
+        with open(f"{extract_path}/slide_{num}.html", "w") as file:
             file.write(html_string)
         num = num + 1
         
-    extract_path = 'slide'
+    
 
     # Create the directory to extract to if it doesn't exist
     os.makedirs(extract_path, exist_ok=True)
 
     # Configuration for imgkit
-    config = imgkit.config(wkhtmltoimage='C:\\Program Files\\wkhtmltopdf\\bin\\wkhtmltoimage.exe')
+    #config = imgkit.config(wkhtmltoimage='C:\\Program Files\\wkhtmltopdf\\bin\\wkhtmltoimage.exe')
 
     # The path to store the images
     image_path = os.path.join(extract_path, 'images')
@@ -271,8 +274,8 @@ def prompt_to_video(video_prompt):
     
     
     # Get list of .jpg and .wav files from the correct directories
-    jpg_files = sorted(glob.glob("slide/images/*.jpg"))
-    wav_files = sorted(glob.glob("slide/*.wav"))
+    jpg_files = sorted(glob.glob(f"{extract_path}/images/*.jpg"))
+    wav_files = sorted(glob.glob(f"{extract_path}/*.wav"))
 
     jpg_files, wav_files
     
@@ -303,12 +306,19 @@ def prompt_to_video(video_prompt):
     # Concatenate all clips into a single video
     video = concatenate_videoclips(clips)
     
-    video_path = "slide/output.mp4"
+    video_path = f"{extract_path}/output.mp4"
 
     # Write the video to a file
     video.write_videofile(video_path)
     
     
     return video_path
-iface = gr.Interface(fn=prompt_to_video, inputs="text", outputs=["file"])
+
+iface = gr.Interface(
+    fn=prompt_to_video, 
+    inputs="text", 
+    outputs=["file"],
+    title="Prompt to Video Tutorial",
+    description="Create a video tutorial to learn about anything!")
+
 iface.launch()