Updated code files

app.py

@@ -59,19 +59,25 @@ with gr.Blocks() as main_app:
     with gr.Row():
         with gr.Column():
             img_input = gr.Image(type="pil", label="Upload an Image")
-            query_input = gr.Textbox(label="Enter query for retrieval", placeholder="Query/Prompt")
+            
             gr.Markdown("""
                         ### Please use this prompt for text extraction 
                         **What text can you identify in this image? Include everything, even if it's partially obscured or in the background.**
                         """)
-            search_input = gr.Textbox(label="Enter search term", placeholder="Search")
+            query_input = gr.Textbox(label="Enter query for retrieval", placeholder="Query/Prompt")
             extract_button = gr.Button("Read Doc!")
+            
+            search_input = gr.Textbox(label="Enter search term", placeholder="Search")
             search_button = gr.Button("Search!")
             
+            
         with gr.Column():
             extracted_text_op = gr.Textbox(label="Output")
             search_text_op = gr.HTML(label="Search Results")
+            
+            download_button = gr.Button("Download Plain Text")
         
+        # Retrieval
         extracted_text_state = gr.State()
         extract_button.click(
             extraction,
@@ -79,10 +85,18 @@ with gr.Blocks() as main_app:
             outputs=[extracted_text_op, extracted_text_state]
         )
         
+        # Search
         search_button.click(
             search_text,
             inputs=[extracted_text_state, search_input],
             outputs=[search_text_op]
         )
         
+        # Download
+        download_button.click(
+            lambda text: gr.File.save_text_to_file(text, "extracted_text.txt"),
+            inputs=[extracted_text_state],
+            outputs=[gr.File(label="Download Extracted Text")]
+        )
+        
 main_app.launch()