Upload app.py

app.py

@@ -0,0 +1,112 @@
+#!/usr/bin/env python
+# coding: utf-8
+
+# <h1>Table of Contents<span class="tocSkip"></span></h1>
+# <div class="toc"><ul class="toc-item"></ul></div>
+
+# Выполнено!
+
+# In[1]:
+
+
+import numpy as np
+import pandas as pd
+import matplotlib.pyplot as plt
+import gradio as gr
+
+
+# In[2]:
+
+
+header_type = np.dtype([('chunkId', 'a4'), ('chunkSize', 'i'), ('format', 'S4'), ('subchunk1Id', 'S4'), ('subchunk1Size', np.int32),
+                   ('audioFormat', np.int16), ('numChannels', np.int16), ('sampleRate', np.int32), ('byteRate', np.int32),
+                   ('blockAlign', np.int16), ('bitsPerSample', np.int16), ('subchunk2Id', 'S', 4), ('subchunk2Size', np.int32), 
+                   ])
+
+
+# In[3]:
+
+
+bmp_header_124 = np.dtype([('type', 'S2'), ('size', np.int32, ), ('reserve', np.int16), ('reserve2', np.int16), ('offset', np.int32),
+                     ('bV5Size', 'I'), ('bV5Width', 'i'), ('bV5Height', 'i'), ('bV5Planes', 'H'), ('bV5BitCount', 'H'),
+                     ('bV5Compression', 'I'), ('bV5SizeImage', 'I'), ('bV5XPelsPerMeter', 'i'), ('bV5YPelsPerMeter', 'i'),
+                      ('bV5ClrUsed', 'I'), ('bV5ClrImportant', 'I'), ('bV5RedMask', 'I'), ('bV5GreenMask', 'I'),
+                       ('bV5BlueMask', 'I'), ('bV5AlphaMask', 'I'), ('bV5CSType', 'I'),
+                       ('bV5Endpoints', '9i'), ('bV5GammaRed', 'I'), ('bV5GammaGreen', 'I'), ('bV5GammaBlue', 'I'),
+                       ('bV5Intent', 'I'), ('bV5ProfileData', 'I'), ('bV5ProfileSize', 'I'), ('bV5Reserved', 'I')])
+bmp_header_40 = np.dtype([('type', 'S2'), ('size', np.int32, ), ('reserve', np.int16), ('reserve2', np.int16), ('offset', np.int32),
+                     ('bV5Size', 'I'), ('bV5Width', 'i'), ('bV5Height', 'i'), ('bV5Planes', 'H'), ('bV5BitCount', 'H'),
+                     ('bV5Compression', 'I'), ('bV5SizeImage', 'I'), ('bV5XPelsPerMeter', 'i'), ('bV5YPelsPerMeter', 'i'),
+                      ('bV5ClrUsed', 'I'), ('bV5ClrImportant', 'I')])
+bmp_dtypes = {124: bmp_header_124, 40: bmp_header_40}
+
+
+# In[4]:
+
+
+filename = r"C:\Users\Ilia\Downloads\DJI.bmp"
+def load_bmp(filename):
+    if isinstance(filename, str):
+        return_data = True
+    else:
+        filename = filename.name
+        return_data = False
+    with open(filename, 'rb') as file:
+        header_type = np.int16(file.read(15)[14])
+        file.seek(0)
+        header = np.fromfile(file, dtype = bmp_dtypes[header_type], count = 1)
+        byte_count_per_pxl = header['bV5BitCount'][0] // 8
+        bmp_colors = np.dtype([('b', np.int32), ('g', np.int32), ('r', np.int32)])
+        colors = np.fromfile(file ,dtype = f'{byte_count_per_pxl}B', count = header['bV5SizeImage'][0])
+        fig, ax = plt.subplots(1, 1)
+        colors[:, [0, 2]] = colors[:, [2, 0]]
+        ax.imshow(np.flip(colors[:, :3].reshape((header['bV5Height'][0], header['bV5Width'][0], 3)), axis = 0))
+        if return_data:
+            return header, colors
+        else:
+            return gr.update(value = fig, visible = True)
+    
+    
+    
+def load_wav(filename):
+    if isinstance(filename, str):
+        return_data = True
+    else:
+        filename = filename.name
+        return_data = False
+    with open(filename, 'rb') as file:
+        header = np.fromfile(file, dtype = header_type, count = 1)
+        data = np.fromfile(file, dtype = np.int16, count = header['subchunk2Size'][0] // 2)\
+                            .reshape(-1, header['numChannels'][0]).T.mean(axis = 0)
+        fig, ax = plt.subplots()
+        ax.plot(data);
+    if return_data:
+        return header, data
+    else:
+        return [gr.update(value = fig, visible = True), gr.update(value = (header[0][7], data), visible = True)];
+
+
+# In[5]:
+
+
+with gr.Blocks() as app:
+    with gr.Row():
+        with gr.Column():
+            wav_file = gr.File(label = 'Загрузите WAV файл', file_count = 'single', file_types = ['.wav'])
+            wav_btn = gr.Button(value = 'Обработать WAV')
+            bmp_file = gr.File(label = 'Загрузите BMP файл', file_count = 'single', file_types = ['.bmp'])
+            bmp_btn = gr.Button(value = 'Обработать BMP')
+            
+        with gr.Column():
+            wav_plot = gr.Plot(visible = False)
+            audio = gr.Audio(visible = False)
+            bmp_plot = gr.Plot(visible = False)
+    wav_btn.click(load_wav, inputs = wav_file, outputs = [wav_plot, audio])  
+    bmp_btn.click(load_bmp, inputs = bmp_file, outputs = [bmp_plot])
+
+
+# In[6]:
+
+
+app.launch(share = True)
+