Update app.py

app.py

@@ -1,32 +1,38 @@
+import gradio as gr
 import numpy as np
-from pydub import AudioSegment
-from matplotlib import pyplot as plt
 from pysstv.color import Robot36
-import gradio as gr
-
-def decode_sstv(audio_file):
-    # Чтение аудиофайла
-    audio = AudioSegment.from_file(audio_file)
-    samples = np.array(audio.get_array_of_samples(), dtype=np.int16)
+from pysstv.utils import load_wav
+from io import BytesIO
+import matplotlib.pyplot as plt
 
-    # Декодирование SSTV сигнала
-    sstv = Robot36(samples, sample_rate=audio.frame_rate)
-    img = sstv.decode()
+def decode_sstv(file):
+    try:
+        # Загрузка WAV файла
+        wav_data, sample_rate = load_wav(file.name)
+        
+        # Используем декодер Robot36
+        sstv = Robot36(wav_data, sample_rate)
+        image = sstv.decode()
 
-    # Отображение изображения
-    plt.imshow(img)
-    plt.axis('off')
-    img_path = 'decoded_image.png'
-    plt.savefig(img_path, bbox_inches='tight', pad_inches=0)
-    return img_path
+        # Конвертируем изображение для отображения
+        fig, ax = plt.subplots()
+        ax.imshow(image)
+        ax.axis('off')
+        buf = BytesIO()
+        plt.savefig(buf, format="png")
+        buf.seek(0)
+        return buf
+    except Exception as e:
+        return f"Ошибка: {e}"
 
+# Интерфейс Gradio
 iface = gr.Interface(
     fn=decode_sstv,
-    inputs=gr.Audio(type="filepath", label="Upload SSTV Audio File"),
-    outputs=gr.Image(label="Decoded Image"),
-    title="SSTV Decoder",
-    description="Upload an SSTV audio file to decode the image."
+    inputs=gr.File(label="Загрузите WAV файл (SSTV)"),
+    outputs=gr.Image(type="file", label="Декодированное изображение"),
+    title="SSTV Декодер",
+    description="Загрузите WAV файл с SSTV сигналом для декодирования изображения."
 )
 
-if __name__ == "__main__":
-    iface.launch()
+# Запуск интерфейса
+iface.launch()