Upload folder using huggingface_hub

app.py

@@ -192,7 +192,7 @@ def generate_job_demo(num_molecules, return_top_k, docking_mode):
     CPU demo generation endpoint.
 
     This does not run live REINVENT generation.
-    It returns a backend-style generation job response using the packaged example rows,
+    It returns a backend-style generation job response using packaged example rows,
     so frontend/filter/backend teams can test JSON + CSV + job history behavior.
     """
     job_id = f"gen_{uuid.uuid4().hex[:8]}"
@@ -232,6 +232,21 @@ def generate_job_demo(num_molecules, return_top_k, docking_mode):
     csv_path = output_dir / f"{job_id}_results.csv"
     df.to_csv(csv_path, index=False)
 
+    job_summary = {
+        "job_id": job_id,
+        "status": "completed",
+        "mode": "generate_job_cpu_demo",
+        "created_at": created_at,
+        "preset": "egfr_generator",
+        "docking_mode": docking_mode,
+        "num_requested": num_molecules,
+        "num_generated": len(rows),
+        "num_returned": len(clean_rows),
+        "download_csv_file": str(csv_path),
+        "live_generation": False,
+        "note": "CPU demo uses packaged example rows. Live REINVENT generation is not executed in this Space version."
+    }
+
     response = {
         "job_id": job_id,
         "status": "completed",
@@ -255,7 +270,7 @@ def generate_job_demo(num_molecules, return_top_k, docking_mode):
         ],
     }
 
-    return df, response, str(csv_path)
+    return job_summary, df, response, str(csv_path)
 
 
 loaded_ok, loaded_msg = download_bundle_light()
@@ -310,16 +325,20 @@ Important:
             """
 This tab demonstrates the expected backend behavior for generation jobs.
 
-It returns:
-
-- job_id
-- status
-- summary
-- results JSON
-- downloadable CSV file
-
 Important:
 This CPU Space does not run live REINVENT generation. It uses packaged example rows to let frontend, filters, and backend test the full response format.
+
+The output is split into 4 parts:
+
+1. Job summary: job-level metadata such as job_id, status, preset, and summary.
+2. Generated results table: molecule-level rows only.
+3. Backend-style generation JSON: full response containing job metadata + results.
+4. Download CSV: downloadable file for the same molecule-level results.
+
+In the real backend:
+- The frontend should consume the JSON.
+- The user can optionally download the CSV.
+- The backend should store job history using job_id/status/created_at.
 """
         )
 
@@ -343,16 +362,18 @@ This CPU Space does not run live REINVENT generation. It uses packaged example r
 
         gen_btn = gr.Button("Generate Job Demo")
 
-        gen_table = gr.Dataframe(label="Generated results table")
-        gen_json = gr.JSON(label="Backend-style generation JSON")
+        gen_summary = gr.JSON(label="Job summary")
+        gen_table = gr.Dataframe(label="Generated molecule results table")
+        gen_json = gr.JSON(label="Full backend-style generation JSON")
         gen_file = gr.File(label="Download CSV")
 
         gen_btn.click(
             generate_job_demo,
             inputs=[gen_num_molecules, gen_return_top_k, gen_docking_mode],
-            outputs=[gen_table, gen_json, gen_file],
+            outputs=[gen_summary, gen_table, gen_json, gen_file],
         )
-    
+
+
     with gr.Tab("Example Generate Response"):
         gr.Markdown(
             """