Update app.py

app.py

@@ -33,15 +33,13 @@ def main():
     samples = np.arange(num_samples)
     signal = np.exp(1j * (freq) * samples)
 
-    dft_result = np.fft.fftshift(np.fft.fft(signal))
-    dft_result = dft_result / np.max(np.abs(dft_result))
-    dft_freqs = np.fft.fftshift(np.fft.fftfreq(len(signal)))
-
     dtft_samples = 2048
-    dtft_result = np.fft.fftshift(np.fft.fft(signal, dtft_samples))
-    dtft_result = dtft_result / np.max(np.abs(dtft_result))
+    dtft_result = np.fft.fftshift(np.fft.fft(signal, dtft_samples)) / num_samples
     dtft_freqs = np.fft.fftshift(np.fft.fftfreq(dtft_samples))
 
+    dft_result = np.fft.fftshift(np.fft.fft(signal)) / num_samples
+    dft_freqs = np.fft.fftshift(np.fft.fftfreq(len(signal)))
+
     dft_freqs_rad = 2 * np.pi * dft_freqs
     dtft_freqs_rad = 2 * np.pi * dtft_freqs