Update app.py

app.py

@@ -253,8 +253,9 @@ def run_script(script: str, output_format: str, dataset_info: Dict[str, Any] = N
                     if not output_file:
                         output_file = file_path
         
-        # Převeď na base64 - vezmi největší soubor (pravděpodobně nejkomplexnější)
+        # Převeď na base64 - sbírej všechny soubory (galerie) a pro kompatibilitu nech i single preview
         data_b64 = ""
+        gallery_b64: list[str] = []
         if all_output_files:
             # Najdi největší soubor
             largest_file = max(all_output_files, key=lambda f: os.path.getsize(f) if os.path.exists(f) else 0)
@@ -264,12 +265,20 @@ def run_script(script: str, output_format: str, dataset_info: Dict[str, Any] = N
                     if len(blob) > max_output:
                         raise HTTPException(status_code=413, detail="Výstup je příliš velký")
                     data_b64 = base64.b64encode(blob).decode("utf-8")
+            # Naplň galerii
+            for fpath in all_output_files:
+                if os.path.exists(fpath):
+                    with open(fpath, "rb") as fb:
+                        blob = fb.read()
+                        if len(blob) <= max_output:
+                            gallery_b64.append(base64.b64encode(blob).decode("utf-8"))
         elif output_file and os.path.exists(output_file):
             with open(output_file, "rb") as f:
                 blob = f.read()
                 if len(blob) > max_output:
                     raise HTTPException(status_code=413, detail="Výstup je příliš velký")
                 data_b64 = base64.b64encode(blob).decode("utf-8")
+                gallery_b64 = [data_b64]
         else:
             # Debug: vypiš všechny soubory v tmp adresáři
             all_files = os.listdir(tmp)
@@ -283,6 +292,7 @@ def run_script(script: str, output_format: str, dataset_info: Dict[str, Any] = N
             "stderr": res.stderr,
             "output_b64": data_b64,
             "all_files": [os.path.basename(f) for f in all_output_files],  # Debug info
+            "gallery_b64": gallery_b64,
         }
 
 
@@ -308,6 +318,7 @@ def advanced_visualization(req: VisualizationRequest) -> Dict[str, Any]:
             },
             "generated_files": result.get("all_files", []),  # Seznam všech vygenerovaných souborů
             "file_count": len(result.get("all_files", [])),  # Počet vygenerovaných souborů
+            "visualizations_multi": result.get("gallery_b64", []),
         }
     except HTTPException:
         raise
@@ -318,4 +329,4 @@ def advanced_visualization(req: VisualizationRequest) -> Dict[str, Any]:
 if __name__ == "__main__":
     import uvicorn
 
-    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", "7860")))
+    uvicorn.run(app, host="0.0.0.0", port=int(os.getenv("PORT", "7860")))