cobol_analysis / app.py
longquan's picture
update app.py 20240128
a008e97 verified
import os
import openai
import json
import tiktoken
import gradio as gr
import time
"""
使用azure openai作为GPT模型
进行cobol代码分析
UI采用gradio框架
UI使用chatbot进行交互
已经实现chatbot的交互问答以及历史记录显示和历史内容保存
chatbot上面不显示prompt内容
实现稳定输出和创造性输出的切换
TODO:
1.还需要一个 stop 生成
2.流式stream输出
3.few-shot learning sample
"""
# 通过max_response_tokens控制回复的长度
max_response_tokens = 8000
history_show = []
temperature=0.5
top_p=0.95
# Load config values
with open('config.json') as config_file:
config_details = json.load(config_file)
# Setting up the deployment name 这个地方不是模型名字,是Azure OpenAI的部署名字
chatgpt_model_name = config_details['CHATGPT_MODEL']
openai.api_type = "azure"
# The API key for your Azure OpenAI resource.
openai.api_key = config_details['OPENAI_API_KEY']
# The base URL for your Azure OpenAI resource. e.g. "https://<your resource name>.openai.azure.com"
openai.api_base = config_details['OPENAI_API_BASE']
# Currently Chat Completions API have the following versions available: 2023-03-15-preview
openai.api_version = config_details['OPENAI_API_VERSION']
def radio_change(choice):
global temperature,top_p
if choice=="安定出力":
temperature=0.5
top_p=0.95
elif choice=="積極出力":
temperature=0.7
top_p=0.95
# Defining a function to send the prompt to the ChatGPT model
# More info : https://learn.microsoft.com/en-us/azure/cognitive-services/openai/how-to/chatgpt?pivots=programming-language-chat-completions
def cobol_analysis_process(history, messages, model_name, max_response_tokens=500):
print("temperature=",temperature,"top_p=",top_p)
response = openai.ChatCompletion.create(
engine=model_name,
messages=messages,
temperature=temperature,
top_p=top_p,
# temperature=0.7,
# top_p=0.95,
max_tokens=max_response_tokens,
frequency_penalty=0,
presence_penalty=0,
# stop="非非",
stream=True,
)
# print("response",response)
print("===========history",history)
history[-1][1] = ""
history_show[-1][1] = ""
for response_ in response:
for choice in response_.choices:
history[-1][1] += choice.delta.content if "content" in choice.delta else ""
history_show[-1][1] += choice.delta.content if "content" in choice.delta else ""
# Defining a function to print out the conversation in a readable format
# def print_conversation(messages):
# for message in messages:
# print(f"[{message['role'].upper()}]")
# print(message['content'])
# print()
def preprocess(history):
# print("history",history)
base_system_message = "あなたは優秀なCOBOLコード分析者です。あなたの仕事は要件に基づいてCOBOLコードを分析し、結果を出力することです。結果は日本語で出力する必要があります。"
messages=[{"role": "system", "content": base_system_message}]
for content in history:
messages.append({"role": "user", "content": content[0]})
if content[1] is not None:
messages.append({"role": "assistant", "content": content[1]})
print("messages",messages)
# response = cobol_analysis_process(messages, chatgpt_model_name, max_response_tokens)
# history[-1] = (history[-1][0], response)
cobol_analysis_process(history,messages, chatgpt_model_name, max_response_tokens)
# print_conversation(messages)
# 点击【提出】按钮后调用greet函数进行处理
def greet(history,user_input,analysis_options):
# print("==========analysis_options=============",analysis_options)
analysis_content=""
if analysis_options=="全体概要 入出力 COPY句 サブルーチン解析":
analysis_content="""
1.概述一下这个程序主要做了什么,全体程序的数据流程以及每个模块的主要内容。全体概要进行说明并使用table表格输出内容。\n
2.程序中所有的入力参数和出力参数,要求使用table表格分别表示,要求每一个对象要有简要的介绍。要再次确认不能有遗漏项目。\n
3.程序中所有的COPY句(COPY文),总结成list表格显示。要求每一个对象要有简要的介绍。要再次确认不能有遗漏项目,所有的COPY句都要总结并在list中输出。\n
4.全体程序中使用的子程序,包括CALL呼叫的子程序,调用外部文件的子程序。这些子程序总结成list表格显示。要求每一个对象要有简要的介绍。要再次确认不能有遗漏项目。
"""
elif analysis_options=="データ定義分析":
analysis_content="""
1.要求分析每一行COBOL代码,不能遗漏任何数据定义行,分析内容使用table表格输出
2.数据定义内容输出格式[等级][项目名][数据类型][长度][初期値]
3.PIC Xデータ型は文字型,PIC 9データ型は数値型
"""
elif analysis_options=="IF ELSE END解析":
analysis_content="""
要求:根据下面的要求以及分析例子分析上面的COBOL代码并使用table表格输出结果
1. 分析每一行cobol代码
2. 分析WHILE语句中条件内容
3. 全部IF ... OR ... ELSE ... END条件语句中条件,变量名,变量数值或者字段内容变化,MOVEコマンド内容,DISPLAY显示的内容,VCALL调用的子程序内容,PERFORM调用内容,RETURN返回内容。这些内容要使用table表格简要表示(tabel列内容包括 [行番号],[コマンド/条件],[層級],[変数名],[変数の変化],[MOVEコマンド内容],[DISPLAY内容],[CALL内容],[PERFORM内容],[RETURN内容])
4. [コマンド/条件]列需要把条件语句的全部内容都写入,条件语句结束标志END和ELSE需要单独一行加入[コマンド/条件]列,嵌套多层IF条件语句中的每一个ELSE,END都不能省略。
5. 程序中注释的语句不需要分析,不需要输出结果
6. 如果有嵌套IF ... ELSE ... IF ... ELSE ... END ... END 需要table中明确表示层级关系
7. 如果是同级别IF ... ELSE ... END table中层级关系数字相同
8. 如果有嵌套 WHILE 需要table中明确表示层级关系
9. CASE OF END语句不要表示[層級]数值
10. 如果是同级别WHILE, table中层级关系数字相同
11. RETURN: S 表示程序终了,在[RETURN内容]列输出[プログラム終了]
12. DISPLAY语句需要把全部内容显示在[DISPLAY内容]列,不能遗漏内容
例:DISPLAY "FMクブン エラー4 HINCODE = " L-HINCODE
输出 '"FMクブン エラー4 HINCODE = " L-HINCODE'
13. [変数の変化]列需要明确表示变数的变化状况。
例:IF: NB-CNT > 0
输出 NB-CNTが0より大きい場合
例:IF: L-FM = "1"
输出 L-FMが1となる場合
"""
elif analysis_options=="TABLE COND ACT END解析":
analysis_content="""
要求分析每一行cobol代码,结果使用table表格显示
如果有嵌套TABLE COND ACT END需要table中明确表示层级关系
同一个TABLE COND ACT END中所有的层级都相同
全部TABLE COND ACT END语句中条件,变量名,判断条件,判断结果。这些内容要使用table表格简要表示(tabel列内容包括 [行番号],[条件],[層級],[変数名],[判断条件],[判断结果])
例:
005070 TABLE: MSKSJ010
005080 COND: MSKSJ010
005090 NHINW-KBN2 (9) = "1" :Y,Y,N,N,N: MSKSJ010
005130 ACT: MSKSJ010
005230 NSKD1-KBN12 := "3" :-,-,-,-,X: MSKSJ010
005240 END: MSKSJ010
[行番号] 005090
[条件] NHINW-KBN2 (9) = "1"
[層級] 1
[変数名] NHINW-KBN2 (9)
[判断条件/変数値変化] "1" かどうかのチェック
[判断結果] :Y,Y,N,N,N:
"""
elif analysis_options=="コード解析":
analysis_content="""分析上面每一行cobol代码,不能有遗漏的代码行,使用table输出结果。table表格的列名[行番号 COBOLコード コード解析結果]
sample 1:
clang0 DS_START_PROC SECTION.
行番号:clang0
COBOLコード:DS_START_PROC SECTION
コード解析結果:DS_START_PROCというセクションの開始を宣言しています。
sample 2:
001120 UNTIL: X = MTOSM2W-KOSU
行番号:001120
COBOLコード:UNTIL: X = MTOSM2W-KOSU
コード解析結果:この行は、XがMTOSM2W-KOSUと等しいまでのループを示しています。
"""
elif analysis_options=="カスタマイズprompt":
analysis_content=""
history_show.append([analysis_options+"\n\n"+user_input, None])
if user_input != "":
user_input = user_input+"\n\n"+analysis_content
else:
user_input = ""
print("user_input==========",user_input)
history.append([user_input, None])
# print("history", history)
preprocess(history)
return history_show, gr.Textbox(value="", interactive=False)
def bot(history_show):
yield history_show
def print_like_dislike(x: gr.LikeData):
print(x.index, x.value, x.liked)
# 页面内容输出控制
with gr.Blocks() as demo:
gr.Markdown("""
<h1 style="text-align: center;">COBOL解析</h1>
""") # 设置标题 可以使用markdown语法
chatbot = gr.Chatbot(
[],
elem_id="chatbot",
bubble_full_width=False,
show_copy_button=True,
avatar_images=(None, (os.path.join(os.path.dirname(__file__), "1.png"))),
)
analysis_options = gr.Dropdown(['全体概要 入出力 COPY句 サブルーチン解析', 'データ定義分析', 'IF ELSE END解析', 'TABLE COND ACT END解析', 'コード解析', 'カスタマイズprompt'], label="解析タイプ選択")
radio=gr.Radio(["安定出力", "積極出力"], label="ランダム性制御", info="「安定出力」を採用するとモデルはより多くの決定論的な応答を生成します。「積極出力」を採用するとより多くの創造的な応答が生じます。")
user_input = gr.Textbox(scale=4,show_label=False,placeholder="user input", container=False,lines=1) # 设置输入框
# 使用gr.ClearButton来清空chatbot记录的内容
clear1 = gr.ClearButton([user_input],value="入力コンテンツクリア")
clear2 = gr.ClearButton([user_input, chatbot],value="Chatコンテンツクリア")
radio.change(fn=radio_change, inputs=radio)
txt_msg = user_input.submit(greet, [chatbot,user_input,analysis_options],[chatbot,user_input], queue=False).then(
bot, chatbot, chatbot, api_name="bot_response"
)
txt_msg.then(lambda: gr.Textbox(interactive=True), None, [user_input], queue=False)
chatbot.like(print_like_dislike, None, None)
demo.queue()
if __name__ == "__main__":
demo.launch(share=True)