Update app.py

app.py

@@ -24,12 +24,13 @@ def get_chapters(epub_file):
 
 def display_chapters(epub_file):
     chapters = get_chapters(epub_file)
-    return "\n".join([f"{title} ({length} characters)" for title, length, _ in chapters])
+    chapter_list = [f"{title} ({length} characters)" for title, length, _ in chapters]
+    return chapter_list, gr.Dropdown.update(choices=chapter_list)
 
 def get_chapter_content(epub_file, selected_chapter):
     chapters = get_chapters(epub_file)
     for title, _, content in chapters:
-        if title in selected_chapter:
+        if f"{title} (" in selected_chapter:  # Match the title part of the dropdown option
             return content
     return "Chapter not found"
 
@@ -40,15 +41,15 @@ def create_interface():
             epub_input = gr.File(label="Upload EPUB File")
             chapters_output = gr.Textbox(label="Available Chapters", interactive=False)
         
-        epub_input.upload(display_chapters, epub_input, chapters_output)
+        chapter_dropdown = gr.Dropdown(label="Select a chapter", choices=[], interactive=True)
         
-        with gr.Row():
-            chapter_select = gr.Textbox(label="Enter the chapter title you want to read")
-            read_button = gr.Button("Read Chapter")
+        epub_input.upload(display_chapters, epub_input, [chapters_output, chapter_dropdown])
+        
+        read_button = gr.Button("Read Chapter")
         
         chapter_content = gr.Textbox(label="Chapter Content", interactive=False)
         
-        read_button.click(get_chapter_content, inputs=[epub_input, chapter_select], outputs=chapter_content)
+        read_button.click(get_chapter_content, inputs=[epub_input, chapter_dropdown], outputs=chapter_content)
     
     return interface