app.py

import gradio as gr
import os
import codecs
import numpy as np
import pandas as pd
from scipy.signal import find_peaks


def process_text(file):

    row = []  # 行号
    x = []    # x值
    y = []    # y值

    filename = file.split(".")[0]
    target_path = f"{filename}.csv"  # 目标路径
    
    # 读取文件内容
    with codecs.open(file, 'r', 'utf-8') as f:
        lines = f.readlines()
        data = []
        for line in lines:
            temp = line.split()
            temp1 = [int(float(x)) for x in temp]
            data.append(temp1)
        data = np.array(data)
        peaks, _ = find_peaks(data[:, 1], distance=2500)  # 根据第2列找到所有峰值， 前后比较阈值为distance=2500

        temp_row = []
        temp_x = []
        temp_y = []
        for peak in peaks:
            temp_row.append(peak)
            temp_x.append(data[peak, 0])
            temp_y.append(data[peak, 1])

        temp_y = sorted(enumerate(temp_y), key=lambda z: z[1], reverse=True)

        for index, value in temp_y:
            row.append(temp_row[index])
            x.append(temp_x[index])
            y.append(value)

        datas = {
                'row_num': row,
                'x': x,
                'y': y,
            }

        df = pd.DataFrame(datas)
        df.to_csv(target_path, index=False)
    
    return df


iface = gr.Interface(
    fn=process_text,
    inputs="file",
    outputs="text",
    title="Upload and Process Text File",
    description="Upload a txt file and it will be processed by this app.",
)

iface.launch()