add examples

.dockerignore

@@ -160,3 +160,5 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+gradio_cached_examples/

.gitignore

@@ -160,3 +160,6 @@ cython_debug/
 #  and can be added to the global gitignore or merged into this file.  For a more nuclear
 #  option (not recommended) you can uncomment the following to ignore the entire idea folder.
 #.idea/
+
+gradio_cached_examples/
+.DS_Store

main.py

@@ -1,6 +1,7 @@
 import io
 import os
 from typing import List, Literal
+from pathlib import Path
 
 import gradio as gr
 from loguru import logger
@@ -78,12 +79,12 @@ def get_mp3(text: str, voice: str, api_key: str = None) -> bytes:
             return file.getvalue()
 
 
-def generate_audio(file: bytes, openai_api_key: str = None) -> bytes:
+def generate_audio(file: str, openai_api_key: str = None) -> bytes:
 
     if not os.getenv("OPENAI_API_KEY", openai_api_key):
         raise gr.Error("OpenAI API key is required")
 
-    reader = PdfReader(io.BytesIO(file))
+    reader = PdfReader(Path(file).open("rb"))
     text = "\n\n".join([page.extract_text() for page in reader.pages])
 
     llm_output = generate_dialogue(text)
@@ -108,6 +109,7 @@ def generate_audio(file: bytes, openai_api_key: str = None) -> bytes:
 
     return audio, transcript
 
+
 description = """
 <p style="text-align:center">
   <strong>Convert any PDF into a podcast episode! Experience research papers, websites, and more in a whole new way.</strong>
@@ -118,10 +120,11 @@ def generate_audio(file: bytes, openai_api_key: str = None) -> bytes:
     title="PDF to Podcast",
     description=description,
     fn=generate_audio,
+    examples=[[p.__fspath__()] for p in Path("examples").glob("*.pdf")],
     inputs=[
         gr.File(
             label="PDF",
-            type="binary",
+            # type="binary",
         ),
         gr.Textbox(
             label="OpenAI API Key",
@@ -135,6 +138,8 @@ def generate_audio(file: bytes, openai_api_key: str = None) -> bytes:
     allow_flagging=False,
     clear_btn=None,
     head=os.getenv("HEAD"),
+    concurrency_limit=20,
+    cache_examples="lazy",
 )
 
 demo.launch(