Update app.py

app.py

@@ -7,6 +7,7 @@ from typing import Tuple, Dict, Any, Optional
 import tempfile
 import io
 import datetime
+import base64
 
 class FeedbackTransformer:
     """
@@ -258,6 +259,31 @@ class FeedbackTransformer:
 
         return analysis_text
 
+    def get_download_data(self, output_format='xlsx'):
+        """
+        Return the data as bytes for direct download (NEW METHOD).
+        """
+        if self.transformed_data is None:
+            raise ValueError("No transformed data to save")
+
+        # Create filename with original filename prefix and timestamp
+        timestamp = datetime.datetime.now().strftime("%Y%m%d_%H%M%S")
+        prefix = self.original_filename if self.original_filename else 'transformed_feedback'
+
+        if output_format == 'xlsx':
+            filename = f"{prefix}_transformed_{timestamp}.xlsx"
+            # Create bytes buffer
+            buffer = io.BytesIO()
+            self.transformed_data.to_excel(buffer, index=False)
+            buffer.seek(0)
+            return buffer.getvalue(), filename
+        else:  # csv
+            filename = f"{prefix}_transformed_{timestamp}.csv"
+            # Create string buffer and encode to bytes
+            buffer = io.StringIO()
+            self.transformed_data.to_csv(buffer, index=False)
+            return buffer.getvalue().encode('utf-8'), filename
+
     def save_transformed_data(self, output_format='xlsx'):
         """
         Save the transformed data and return the file path.
@@ -379,6 +405,7 @@ def process_file(file_obj, topic_prefix, sentiment_prefix, category_prefix,
                 text_column, recommendation_column, output_format, analyze_data, selected_columns):
     """
     Main processing function for Gradio interface.
+    Returns both file path and download data for multiple download options.
     """
     try:
         # Extract actual column names from display format
@@ -423,15 +450,58 @@ def process_file(file_obj, topic_prefix, sentiment_prefix, category_prefix,
         if analyze_data:
             analysis_result = transformer.analyze_data()
 
-        # Save transformed data
+        # Save transformed data (for regular file download)
         output_file = transformer.save_transformed_data(output_format)
-        status_msg += f"\n💾 File saved successfully: {os.path.basename(output_file)}\n"
+        
+        # Get download data (for direct download button)
+        download_data, filename = transformer.get_download_data(output_format)
+        
+        status_msg += f"\n💾 File ready for download: {filename}\n"
 
-        return status_msg, analysis_result, output_file
+        return status_msg, analysis_result, output_file, (download_data, filename)
 
     except Exception as e:
         error_msg = f"❌ Error: {str(e)}\n\n{traceback.format_exc()}"
-        return error_msg, "", None
+        return error_msg, "", None, None
+
+
+def create_download_link(download_data_tuple):
+    """
+    Create a direct download link using data URI (NEW FUNCTION).
+    """
+    if download_data_tuple is None:
+        return None
+    
+    download_data, filename = download_data_tuple
+    
+    # Encode data as base64
+    b64_data = base64.b64encode(download_data).decode()
+    
+    # Determine MIME type
+    if filename.endswith('.xlsx'):
+        mime_type = 'application/vnd.openxmlformats-officedocument.spreadsheetml.sheet'
+    else:
+        mime_type = 'text/csv'
+    
+    # Create data URI
+    data_uri = f"data:{mime_type};base64,{b64_data}"
+    
+    # Return HTML with download link
+    download_html = f"""
+    <div style="text-align: center; padding: 20px; border: 2px dashed #007bff; border-radius: 10px; background-color: #f8f9fa;">
+        <h3 style="color: #007bff; margin-bottom: 15px;">📥 Download Ready!</h3>
+        <a href="{data_uri}" download="{filename}" 
+           style="display: inline-block; padding: 12px 24px; background-color: #007bff; color: white; 
+                  text-decoration: none; border-radius: 5px; font-weight: bold; font-size: 16px;">
+            📄 Download {filename}
+        </a>
+        <p style="margin-top: 15px; color: #6c757d; font-size: 14px;">
+            Click the button above to download your transformed data file.
+        </p>
+    </div>
+    """
+    
+    return download_html
 
 
 # Create Gradio interface
@@ -453,7 +523,7 @@ with gr.Blocks(title="Feedback Topic & Sentiment Transformer", css="""
     2. Select which original columns to include in the output
     3. Configure column prefixes (or use defaults)
     4. Click "Transform Data" to process
-    5. Download the transformed file
+    5. Download the transformed file using the download button below
     """)
 
     with gr.Row():
@@ -537,16 +607,30 @@ with gr.Blocks(title="Feedback Topic & Sentiment Transformer", css="""
                 label="Data Analysis"
             )
 
-    # Download section - Modified for better download functionality
+    # Download sections - Multiple options for better compatibility
     with gr.Row():
         with gr.Column():
             gr.Markdown("### �� 5. Download Transformed File")
+            
+            # Primary download method - Direct download link
+            download_link = gr.HTML(
+                label="Direct Download",
+                value="Process a file to generate download link",
+                visible=True
+            )
+            
+        with gr.Column():
+            gr.Markdown("### 📥 Alternative Download")
+            # Fallback download method - Traditional file component
             output_file = gr.File(
-                label="Transformed File",
+                label="Fallback Download (if direct download doesn't work)",
                 interactive=False,
                 visible=True
             )
 
+    # Hidden state to store download data
+    download_data_state = gr.State()
+
     # Event handlers
     input_file.change(
         fn=get_column_selector,
@@ -567,7 +651,11 @@ with gr.Blocks(title="Feedback Topic & Sentiment Transformer", css="""
             analyze_checkbox,
             column_selector
         ],
-        outputs=[status_output, analysis_output, output_file]
+        outputs=[status_output, analysis_output, output_file, download_data_state]
+    ).then(
+        fn=create_download_link,
+        inputs=[download_data_state],
+        outputs=[download_link]
     )
 
     # Examples section
@@ -589,7 +677,8 @@ with gr.Blocks(title="Feedback Topic & Sentiment Transformer", css="""
     - Use the numbered column list to easily identify and select columns
     - The text and recommendation column names in configuration are now for reference only
     - To include them in output, select them using the column checkboxes
-    - Click on the download button that appears after processing to download the file
+    - **Two download options provided**: Try the direct download button first. If it doesn't work in your embedded environment, use the fallback download link.
+    - The direct download creates a clickable button that should work better in embedded applications
     """)
 
 # Launch the app