Update app.py

app.py

@@ -236,25 +236,30 @@ def process_woocommerce_data_in_memory(netcom_file):
 
     return output_buffer
 
-def process_file_and_return_csv(uploaded_file):
+def process_woocommerce_data_in_memory(netcom_file):
     """
-    - Takes the uploaded file,
-    - Processes it,
-    - Writes the CSV to a temp file,
-    - Returns that path for Gradio to provide as a downloadable file.
+    Reads the uploaded NetCom CSV file in-memory, processes it to the WooCommerce format,
+    and returns the resulting CSV as bytes, suitable for download.
     """
-    processed_csv_io = process_woocommerce_data_in_memory(uploaded_file)
-
-    with tempfile.NamedTemporaryFile(delete=False, suffix=".csv") as tmp:
-        tmp.write(processed_csv_io.getvalue())
-        tmp.flush()  # ensure data is written to disk
-        temp_path = tmp.name
+    # [Keep all your existing processing code exactly the same until the end]
+    
+    # 9. Convert to CSV (in memory)
+    output_buffer = BytesIO()
+    woo_final_df.to_csv(output_buffer, index=False, encoding='utf-8-sig')
+    output_buffer.seek(0)
+    
+    return output_buffer
 
-    return temp_path
+def process_file(uploaded_file):
+    """
+    Takes the uploaded file, processes it, and returns the CSV as a file-like object
+    """
+    processed_csv_io = process_woocommerce_data_in_memory(uploaded_file)
+    return processed_csv_io
 
-app = gr.Interface(
-    fn=process_file_and_return_csv,
-    inputs=gr.File(label="Upload NetCom CSV", file_types=["text", "csv"]),
+interface = gr.Interface(
+    fn=process_file,
+    inputs=gr.File(label="Upload NetCom CSV", file_types=[".csv"]),
     outputs=gr.File(label="Download WooCommerce CSV"),
     title="NetCom to WooCommerce CSV Processor",
     description="Upload your NetCom Reseller Schedule CSV to generate the WooCommerce import-ready CSV."
@@ -262,4 +267,6 @@ app = gr.Interface(
 
 if __name__ == "__main__":
     openai_api_key = os.getenv("OPENAI_API_KEY")
-    app.launch()
+    if not openai_api_key:
+        print("Warning: OPENAI_API_KEY environment variable not set")
+    interface.launch()