Update app.py

app.py

@@ -7,7 +7,6 @@ from typing import Tuple, Dict, Any, Optional
 import tempfile
 import io
 import datetime
-import base64
 
 class FeedbackTransformer:
     """
@@ -259,31 +258,6 @@ 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.
@@ -405,7 +379,6 @@ 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
@@ -450,58 +423,15 @@ def process_file(file_obj, topic_prefix, sentiment_prefix, category_prefix,
         if analyze_data:
             analysis_result = transformer.analyze_data()
 
-        # Save transformed data (for regular file download)
+        # Save transformed data
         output_file = transformer.save_transformed_data(output_format)
-        
-        # 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"
+        status_msg += f"\n💾 File saved successfully: {os.path.basename(output_file)}\n"
 
-        return status_msg, analysis_result, output_file, (download_data, filename)
+        return status_msg, analysis_result, output_file
 
     except Exception as e:
         error_msg = f"❌ Error: {str(e)}\n\n{traceback.format_exc()}"
-        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
+        return error_msg, "", None
 
 
 # Create Gradio interface
@@ -523,7 +453,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 using the download button below
+    5. Download the transformed file
     """)
 
     with gr.Row():
@@ -607,30 +537,16 @@ with gr.Blocks(title="Feedback Topic & Sentiment Transformer", css="""
                 label="Data Analysis"
             )
 
-    # Download sections - Multiple options for better compatibility
+    # Download section - Modified for better download functionality
     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="Fallback Download (if direct download doesn't work)",
+                label="Transformed File",
                 interactive=False,
                 visible=True
             )
 
-    # Hidden state to store download data
-    download_data_state = gr.State()
-
     # Event handlers
     input_file.change(
         fn=get_column_selector,
@@ -651,11 +567,7 @@ with gr.Blocks(title="Feedback Topic & Sentiment Transformer", css="""
             analyze_checkbox,
             column_selector
         ],
-        outputs=[status_output, analysis_output, output_file, download_data_state]
-    ).then(
-        fn=create_download_link,
-        inputs=[download_data_state],
-        outputs=[download_link]
+        outputs=[status_output, analysis_output, output_file]
     )
 
     # Examples section
@@ -677,8 +589,7 @@ 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
-    - **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
+    - Click on the download button that appears after processing to download the file
     """)
 
 # Launch the app