gradio可视化对话框（）

from transformers import AutoModel, AutoTokenizer
import gradio as gr
import mdtex2html
# -*- coding: utf-8 -*-
import os
import torch

import interfaceAll
from datetime import datetime
import time
import inference_LLaVa

import shutil
import os
import re
import sys
from PIL import Image
from pathlib import Path

# sys.path.insert(0,'/home/notebook/data/group/menglingxuan/project/multimodal_summary')
sys.path.insert(0, 'path/文件路径')
os.environ['http_proxy'] = 'http://nbproxy.mlp.oppo.local:8888'
os.environ['https_proxy'] = 'http://nbproxy.mlp.oppo.local:8888'
from lmm_0119.lmm_test.qwen_vl_summary import *
import interfaceAll
import gradio as gr
from PIL import Image
import Llama_vicuna

from markdown import markdown

# 模型大体名称
listFile_package = {}
listPackageName = []
listFileName = []
# 非语言模型
not_work = ['BelleGroup', 'Bert', 'Bloom', 'BlueLM', 'Clip', 'GPT', 'Mamba', 'OpenFlamingo', 'RedPajama', 'StableLM',
            'T5', 'Xlnet', 'Yuan', 'deepseek', 'tigerbot']
# 传参是否需要img
img_work = ['LLaVA', 'ChatGLM', 'Llama', 'MiniGPT4', 'TinyLlama', 'Intern', 'Qwen']
count = 1
pathSingle = ""
userMOW_input = ""
fileImg = "imageDir"
fileTxt = "txtDirs"
img_style = ['.png', '.jpg', '.gif']
img_paths_list = []
txt_paths_list = []
global isInitSummary, isInitChat
isInitSummary = False
isInitChat = False


# 获取指定目录下所有文件路径（包含子目录中文件路径）
def getDirAllFileNames(strDirPath):
    for filesOne in os.listdir(strDirPath):  # 不仅仅是文件，当前目录下的文件夹也会被认为遍历到
        if ((filesOne.endswith("-models-dirs")) | (filesOne.endswith("-model-dirs"))):
            if (filesOne.endswith("-models-dirs")):
                if filesOne[0:-12] not in not_work:
                    listPackageName.append(filesOne[0:-12])
                    listFile_package[filesOne[0:-12]] = filesOne
            elif (filesOne.endswith("-model-dirs")):
                if filesOne[0:-11] not in not_work:
                    listPackageName.append(filesOne[0:-11])
                    listFile_package[filesOne[0:-11]] = filesOne
    return listPackageName
    # 获取指定目录下所有文件路径（包含子目录中文件路径）


def getMkDirFileNames(strDirPath):
    listFileName = []
    for filessecond in os.listdir(strDirPath):
        # 子目录遍历
        if os.path.isdir(strDirPath + "/" + filessecond):
            listFileName.append(filessecond)
    return listFileName


pathAll = "path/文件路径"
listmodelDirs = getDirAllFileNames(pathAll)
# 初次加载对话模型
interfaceModel = interfaceAll


# 对话结果解析
def parse_text(text):
    """copy from https://github.com/GaiZhenbiao/ChuanhuChatGPT/"""
    lines = text.split("\n")
    lines = [line for line in lines if line != ""]
    count = 0
    for i, line in enumerate(lines):
        if "```" in line:
            count += 1
            items = line.split('`')
            if count % 2 == 1:
                lines[i] = f'
'
            else:
                lines[i] = f'
'
        else:
            if i > 0:
                if count % 2 == 1:
                    line = line.replace("`&#