Update app.py

app.py

@@ -80,20 +80,9 @@ class SyntheticDataGenerator:
         except Exception as e:
             return None, f"Synthetic data generation failed with error: {str(e)}"
 
-    # ---- Report helpers ----
-    def get_quality_report_text(self) -> str:
-        """Return a concise status about the report."""
-        if not self.generator:
-            return "No trained generator available. Please train a model first."
-        try:
-            _ = self.generator.reports(display=False)  # builds report internally
-            return "Quality report generated. Use the button to download."
-        except Exception as e:
-            return f"Failed to generate quality report: {str(e)}"
-
     def get_quality_report_file(self) -> Optional[str]:
         """
-        Generate/export the report and return a file path for download.
+        Generate/export the quality report and return a file path for download.
         Tries to find an existing ZIP; otherwise saves a TXT fallback.
         """
         if not self.generator:
@@ -194,19 +183,6 @@ def generate_data(size: int) -> Tuple[Optional[pd.DataFrame], str]:
     return synthetic_df, f"{status}: {message}"
 
 
-def get_quality_report_and_file():
-    """
-    Return (status_text, download_component_update)
-    The second value updates the DownloadButton with the file path and visibility.
-    """
-    status = generator.get_quality_report_text()
-    path = generator.get_quality_report_file()
-    if path:
-        return status, gr.update(value=path, visible=True)
-    else:
-        return status, gr.update(visible=False)
-
-
 def create_comparison_plot(original_df: pd.DataFrame, synthetic_df: pd.DataFrame) -> Optional[go.Figure]:
     if original_df is None or synthetic_df is None:
         return None
@@ -311,11 +287,9 @@ def create_interface():
                     train_btn = gr.Button("Train Model", variant="primary")
                 with gr.Column():
                     train_status = gr.Textbox(label="Training Status", interactive=False)
-                    quality_report = gr.Textbox(label="Quality Report", lines=8, interactive=False)
 
             with gr.Row():
-                get_report_btn = gr.Button("Get Quality Report", variant="secondary")
-                report_download_btn = gr.DownloadButton("Download Quality Report", visible=False)
+                get_report_btn = gr.DownloadButton("Get Quality Report", variant="secondary")
 
         with gr.Tab("Generate Data"):
             gr.Markdown("### Generate synthetic data from your trained model")
@@ -331,31 +305,6 @@ def create_interface():
                 download_btn = gr.DownloadButton("Download CSV", variant="secondary")
                 comparison_plot = gr.Plot(label="Data Comparison")
 
-        # README footer
-        gr.Markdown(
-            """
-        **Modes of operation:**
-        - **LOCAL mode** trains and generates synthetic data on your own compute resources.
-        - **CLIENT mode** connects to a remote MOSTLY AI platform for training and generation.
-        - Generators trained locally can be imported to the platform for sharing and collaboration.
-        
-        **Key resources managed by the SDK:**
-        - **Generators**: Train on your tabular or language data assets.
-        - **Synthetic datasets**: Generate any number of synthetic samples as needed.
-        - **Connectors**: Connect to organizational data sources for reading and writing data.
-        
-        **Common intents and API primitives:**
-        - Train a generator: `g = mostly.train(config)`
-        - Generate records: `sd = mostly.generate(g, config)`
-        - Probe generator: `df = mostly.probe(g, config)`
-        - Connect to data source: `c = mostly.connect(config)`
-
-        The open source Synthetic Data SDK by MOSTLY AI powers the MOSTLY AI Platform and MOSTLY AI Assistant.
-        
-        Sign up for free and try the [MOSTLY AI Platform](https://app.mostly.ai/) today!
-        """
-        )
-
         # ---- Event handlers ----
         init_btn.click(initialize_sdk, outputs=[init_status])
 
@@ -365,11 +314,8 @@ def create_interface():
             outputs=[train_status],
         )
 
-        # Build + expose quality report for download
-        get_report_btn.click(
-            get_quality_report_and_file,
-            outputs=[quality_report, report_download_btn],
-        )
+        # Direct download of quality report
+        get_report_btn.click(generator.get_quality_report_file, outputs=[get_report_btn])
 
         # Generate data
         generate_btn.click(generate_data, inputs=[gen_size], outputs=[synthetic_data, gen_status])