From dff011358c255164d1d91b182f52af29b8876a20 Mon Sep 17 00:00:00 2001
From: Reckless_Satoshi <reckless.satoshi@protonmail.com>
Date: Sun, 30 Jan 2022 12:07:13 -0800
Subject: [PATCH] Add close button on info dialog

---
 frontend/src/components/InfoDialog.js  | 5 ++++-
 frontend/src/components/UserGenPage.js | 2 +-
 2 files changed, 5 insertions(+), 2 deletions(-)

diff --git a/frontend/src/components/InfoDialog.js b/frontend/src/components/InfoDialog.js
index eca99f13..94b5f45e 100644
--- a/frontend/src/components/InfoDialog.js
+++ b/frontend/src/components/InfoDialog.js
@@ -1,5 +1,5 @@
 
-import {Typography, DialogTitle,  DialogContent, DialogContentText, Button } from "@mui/material"
+import {Typography, DialogActions,  DialogContent, Button } from "@mui/material"
 import React, { Component } from 'react'
 
 export default class InfoDialog extends Component {
@@ -121,6 +121,9 @@ export default class InfoDialog extends Component {
               RoboSats</i> will definitely never ask for your robot token.
             </p>
           </Typography>
+          <DialogActions>
+            <Button onClick={this.props.handleCloseInfo}>Close</Button>
+          </DialogActions>
         </DialogContent>
       </div>
     )
diff --git a/frontend/src/components/UserGenPage.js b/frontend/src/components/UserGenPage.js
index 8873ebd2..93752106 100644
--- a/frontend/src/components/UserGenPage.js
+++ b/frontend/src/components/UserGenPage.js
@@ -120,7 +120,7 @@ export default class UserGenPage extends Component {
         aria-describedby="info-dialog-description"
         scroll="paper"
       >
-        <InfoDialog/>
+        <InfoDialog handleCloseInfo = {this.handleCloseInfo}/>
       </Dialog>
     )
   }