From 02a150d135d962bf84b5aea60a913240ad955c64 Mon Sep 17 00:00:00 2001
From: Rohit <rohit.rajan031101@gmail.com>
Date: Wed, 30 Apr 2025 21:02:38 +0530
Subject: [PATCH] feat: paginate capture screenshots

---
 src/components/run/RunContent.tsx | 103 +++++++++++++++++++++++++++++-
 1 file changed, 100 insertions(+), 3 deletions(-)

diff --git a/src/components/run/RunContent.tsx b/src/components/run/RunContent.tsx
index c3166720..f0074adb 100644
--- a/src/components/run/RunContent.tsx
+++ b/src/components/run/RunContent.tsx
@@ -57,6 +57,9 @@ export const RunContent = ({ row, currentLog, interpretationInProgress, logEndRe
   const [listColumns, setListColumns] = useState<string[][]>([]);
   const [currentListIndex, setCurrentListIndex] = useState<number>(0);
 
+  const [screenshotKeys, setScreenshotKeys] = useState<string[]>([]);
+  const [currentScreenshotIndex, setCurrentScreenshotIndex] = useState<number>(0);
+
   const [expandedView, setExpandedView] = useState<string | null>(null);
 
   const [legacyData, setLegacyData] = useState<any[]>([]);
@@ -90,6 +93,13 @@ export const RunContent = ({ row, currentLog, interpretationInProgress, logEndRe
     }
   }, [row.serializableOutput]);
 
+  useEffect(() => {
+    if (row.binaryOutput && Object.keys(row.binaryOutput).length > 0) {
+      setScreenshotKeys(Object.keys(row.binaryOutput));
+      setCurrentScreenshotIndex(0);
+    }
+  }, [row.binaryOutput]);
+
   const processLegacyData = (legacyOutput: Record<string, any>) => {
     let allData: any[] = [];
 
@@ -240,6 +250,14 @@ export const RunContent = ({ row, currentLog, interpretationInProgress, logEndRe
     }
   };
 
+  const navigateScreenshots = (direction: 'next' | 'prev') => {
+    if (direction === 'next' && currentScreenshotIndex < screenshotKeys.length - 1) {
+      setCurrentScreenshotIndex(currentScreenshotIndex + 1);
+    } else if (direction === 'prev' && currentScreenshotIndex > 0) {
+      setCurrentScreenshotIndex(currentScreenshotIndex - 1);
+    }
+  };
+
   const renderDataTable = (
     data: any[],
     columns: string[],
@@ -789,11 +807,90 @@ export const RunContent = ({ row, currentLog, interpretationInProgress, logEndRe
             </Box>
           )}
 
-          {/* {hasScreenshots && (
+          {hasScreenshots && (
             <>
-              {renderPaginatedScreenshots()}
+              <Accordion defaultExpanded sx={{ mb: 2 }}>
+                <AccordionSummary
+                  expandIcon={<ExpandMoreIcon />}
+                  aria-controls="screenshot-content"
+                  id="screenshot-header"
+                >
+                  <Box sx={{ display: 'flex', alignItems: 'center' }}>
+                    <ImageIcon sx={{ color: '#FF00C3' }} />
+                    <Typography variant='h6' sx={{ ml: 2 }}>
+                      {t('run_content.captured_screenshot.title', 'Screenshots')}
+                    </Typography>
+                  </Box>
+                </AccordionSummary>
+                <AccordionDetails>
+                  <Box sx={{ display: 'flex', alignItems: 'center', justifyContent: 'space-between', mb: 2 }}>
+                    <Button
+                      startIcon={<DownloadIcon />}
+                      href={row.binaryOutput[screenshotKeys[currentScreenshotIndex]]}
+                      download={screenshotKeys[currentScreenshotIndex]}
+                      sx={{ borderColor: '#FF00C3', color: '#FF00C3', borderRadius: 1 }}
+                      variant="outlined"
+                    >
+                      {t('run_content.captured_screenshot.download', 'Download')}
+                    </Button>
+                    
+                    {screenshotKeys.length > 1 && (
+                      <ButtonGroup size="small">
+                        <Button
+                          onClick={() => navigateScreenshots('prev')}
+                          disabled={currentScreenshotIndex === 0}
+                          sx={{
+                            borderColor: '#FF00C3',
+                            color: currentScreenshotIndex === 0 ? 'gray' : '#FF00C3',
+                            '&.Mui-disabled': {
+                              borderColor: 'rgba(0, 0, 0, 0.12)'
+                            }
+                          }}
+                        >
+                          <ArrowBackIcon />
+                        </Button>
+                        <Button
+                          onClick={() => navigateScreenshots('next')}
+                          disabled={currentScreenshotIndex === screenshotKeys.length - 1}
+                          sx={{
+                            borderColor: '#FF00C3',
+                            color: currentScreenshotIndex === screenshotKeys.length - 1 ? 'gray' : '#FF00C3',
+                            '&.Mui-disabled': {
+                              borderColor: 'rgba(0, 0, 0, 0.12)'
+                            }
+                          }}
+                        >
+                          <ArrowForwardIcon />
+                        </Button>
+                      </ButtonGroup>
+                    )}
+                  </Box>
+                  
+                  <Box sx={{ mt: 1 }}>
+                    {screenshotKeys.length > 1 && (
+                      <Chip
+                        label={`Screenshot ${currentScreenshotIndex + 1} of ${screenshotKeys.length}`}
+                        size="small"
+                        sx={{ backgroundColor: '#FF00C3', color: 'white', mb: 2 }}
+                      />
+                    )}
+                    <Box>
+                      <img
+                        src={row.binaryOutput[screenshotKeys[currentScreenshotIndex]]}
+                        alt={`Screenshot ${screenshotKeys[currentScreenshotIndex]}`}
+                        style={{
+                          maxWidth: '100%',
+                          height: 'auto',
+                          border: '1px solid #e0e0e0',
+                          borderRadius: '4px'
+                        }}
+                      />
+                    </Box>
+                  </Box>
+                </AccordionDetails>
+              </Accordion>
             </>
-          )} */}
+          )}
         </TabPanel>
       </TabContext>
     </Box>