astrodatagui: FlaredetectorWidget: show folded fit

main/astrodatagui/ui/FlaredetectorWidget.py

@@ -4,13 +4,13 @@ from matplotlib.figure import Figure
 from matplotlib.backends.backend_qtagg import (
      FigureCanvas, NavigationToolbar2QT as NavigationToolbar)
 
-from astropy.io import fits
 from astropy.visualization import time_support
 import lightkurve as lk
 
 from ...flaredetector.util import (findMaxIndices, plotFlarePeaks,
                                    plotFlareFits, markFlare,
-                                   convertStarndardIndexToFoldedIndex)
+                                   convertStarndardIndexToFoldedIndex,
+                                   getFoldedBestFit)
 from ...flaredetector.flaredetector import calculateFlareFitsForLightcurve
 
 class FlaredetectorWidget(QtWidgets.QWidget):
@@ -254,6 +254,8 @@ class FlaredetectorWidget(QtWidgets.QWidget):
                 cycle, foldedIndex = convertStarndardIndexToFoldedIndex(lc, peak["StandardIndex"])
                 self.figureAxis.plot(lc.phase[lc.cycle == cycle][foldedIndex].value,
                                      lc.flux[lc.cycle == cycle][foldedIndex], "x", color="red")
+            phase, sineFit = getFoldedBestFit(lc)
+            self.figureAxis.plot(phase, sineFit, color="red")
         elif(self.PeriodogramState["Enabled"]):
             lc.plot(label=label, ax=self.figureAxis, view=self.PeriodogramState["View"])
             if(self.PeriodogramState["View"] == "period"):