portalniy-dev commited on
Commit
f72ed58
·
verified ·
1 Parent(s): e098296

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -7
app.py CHANGED
@@ -1,17 +1,23 @@
1
  import gradio as gr
2
  import numpy as np
3
  from pysstv.color import Robot36
4
- from pysstv.utils import load_wav
5
- from io import BytesIO
6
  import matplotlib.pyplot as plt
 
7
 
8
  def decode_sstv(file):
9
  try:
10
- # Загрузка WAV файла
11
- wav_data, sample_rate = load_wav(file.name)
12
-
13
- # Используем декодер Robot36
14
- sstv = Robot36(wav_data, sample_rate)
 
 
 
 
 
 
15
  image = sstv.decode()
16
 
17
  # Конвертируем изображение для отображения
 
1
  import gradio as gr
2
  import numpy as np
3
  from pysstv.color import Robot36
4
+ import wave
 
5
  import matplotlib.pyplot as plt
6
+ from io import BytesIO
7
 
8
  def decode_sstv(file):
9
  try:
10
+ # Открываем WAV файл
11
+ with wave.open(file.name, 'rb') as wav_file:
12
+ n_channels, sampwidth, framerate, n_frames, comptype, compname = wav_file.getparams()
13
+ if n_channels != 1:
14
+ return "Ошибка: WAV файл должен быть монофоническим."
15
+
16
+ # Читаем данные WAV файла
17
+ wav_data = np.frombuffer(wav_file.readframes(n_frames), dtype=np.int16)
18
+
19
+ # Декодируем SSTV
20
+ sstv = Robot36(wav_data, framerate)
21
  image = sstv.decode()
22
 
23
  # Конвертируем изображение для отображения