Update app.py

app.py

@@ -244,7 +244,7 @@ def send_to_model_impl(prompt, model_selection, hf_model_choice, hf_custom_model
         with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
             f.write(summary)
             
-        return summary, [f.name]
+        return summary, download_file # Return the file for download_summary
     except Exception as e:
         error_msg = f"Error processing request: {str(e)}"
         logging.error(error_msg)
@@ -456,7 +456,7 @@ with gr.Blocks(css="""
                         max_lines=50,
                         show_copy_button=True
                     )
-                    download_full_text = gr.Button("📥 Download Full Text")
+                    download_full_text = gr.File(label="📥 Download Full Text")
 
         # Tab 2: Snippet Selection
         with gr.Tab("2️⃣ Snippet Selection"):
@@ -486,9 +486,8 @@ with gr.Blocks(css="""
                 )
                     
                     with gr.Row():
-                        copy_prompt_button = gr.Button("📋 Copy Prompt")
-                        download_prompt = gr.Button("📥 Download Prompt")
-                        download_snippet = gr.Button("📥 Download Selected Snippet")
+                        download_prompt = gr.File(label="📥 Download Prompt")
+                        download_snippet = gr.File(label="📥 Download Selected Snippet")
 
         # Tab 3: Model Processing
         with gr.Tab("3️⃣ Model Processing"):
@@ -552,8 +551,7 @@ with gr.Blocks(css="""
                     )
                     
                     with gr.Row():
-                        copy_summary_button = gr.Button("📋 Copy Summary")
-                        download_summary = gr.Button("📥 Download Summary")
+                        download_summary = gr.File(label="📥 Download Summary")
             
     # Hidden components for file handling
     download_files = gr.Files(label="📥 Downloads", visible=False)
@@ -675,7 +673,8 @@ with gr.Blocks(css="""
                 formatted_text,
                 snippets_list,
                 gr.update(choices=update_snippet_choices(snippets_list), value="Snippet 1 of " + str(len(snippets_list))),
-                [download_file]
+                download_file # Return the file for download_full_text
+                #[download_file]
             )
             
         except Exception as e:
@@ -690,29 +689,25 @@ with gr.Blocks(css="""
                 None
             )
 
-    def handle_snippet_selection(choice, snippets_list):
-        """Handle snippet selection and update prompt"""
+    def handle_snippet_selection(choice, snippets_list): # Add download_snippet output
+        """Handle snippet selection, update prompt, and provide snippet download."""
         if not snippets_list:
-            return (
-                "No snippets available.",  # progress_status
-                "",                       # generated_prompt
-                None                      # download_files
-            )
-            
+            return "No snippets available.", "", None  # Return None for download
+
         try:
             idx = get_snippet_index(choice)
             selected_snippet = snippets_list[idx]
-            
-            # Create downloadable snippet
+
             with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
                 f.write(selected_snippet)
-                
+                snippet_download_file = f.name  # Store the file path
+
             return (
                 f"Selected snippet {idx + 1}",
                 selected_snippet,
-                [f.name]
+                snippet_download_file # Return file for download
             )
-            
+
         except Exception as e:
             error_msg = f"Error selecting snippet: {str(e)}"
             logging.error(error_msg)
@@ -737,8 +732,10 @@ with gr.Blocks(css="""
             # Save prompt for download
             with tempfile.NamedTemporaryFile(delete=False, mode='w', suffix='.txt') as f:
                 f.write(prompt)
+                download_file = f.name
                 
-            return "Prompt generated!", prompt, [f.name]
+            return "Prompt generated!", prompt, download_file # Return the file for download_prompt
+
         except Exception as e:
             logging.error(f"Error generating prompt: {e}")
             return f"Error: {str(e)}", "", None
@@ -754,35 +751,20 @@ with gr.Blocks(css="""
     process_button.click(
         handle_pdf_process,
         inputs=[pdf_input, format_type, context_size],
-        outputs=[  # List of outputs, not dict
-            progress_status,
-            processed_text,
-            pdf_content,
-            snippets,
-            snippet_selector,
-            download_files
-        ]
+        outputs=[progress_status, processed_text, pdf_content, snippets, snippet_selector, download_full_text]
     )
 
     generate_prompt_btn.click(
         handle_prompt_generation,
         inputs=[generated_prompt, custom_prompt, snippet_selector, snippets],
-        outputs=[
-            progress_status,
-            generated_prompt,
-            download_files
-        ]
+        outputs=[progress_status, generated_prompt, download_prompt]
     )
 
     # Snippet handling
     snippet_selector.change(
         handle_snippet_selection,
         inputs=[snippet_selector, snippets],
-        outputs=[
-            progress_status,
-            generated_prompt,
-            download_files
-        ]
+        outputs=[progress_status, generated_prompt, download_snippet] # Connect download_snippet
     )
 
     # Model selection
@@ -809,18 +791,6 @@ with gr.Blocks(css="""
         outputs=[context_size]
     )
 
-    # Download handlers
-    copy_prompt_button.click(
-        fn=None,
-        _js=copy_text_js("generated_prompt"), # Use the helper function
-        outputs=progress_status
-    )
-    copy_summary_button.click(
-        fn=None,
-        _js=copy_text_js("summary_output"), # Use the helper function
-        outputs=progress_status
-    )
-
     def download_file(content: str, prefix: str) -> List[str]:
         if not content:
             return []
@@ -854,10 +824,7 @@ with gr.Blocks(css="""
             groq_api_key,
             openai_api_key
         ],
-        outputs=[
-            summary_output,
-            download_files
-        ]
+        outputs=[summary_output, download_summary]
     )
 
     groq_refresh_btn.click(