Update app.py

app.py

@@ -11,21 +11,33 @@ api_key = os.environ.get("OPENAI_API_KEY")
 client = OpenAI(api_key=api_key)
 
 # Define the model to be used
-MODEL = "gpt-4o"
+MODEL = os.environ.get("MODEL")
 
 def read(filename):
     with open(filename) as f:
         data = f.read()
     return data
     
-sys_prompt = read('system_prompt.txt')
+SYS_PROMPT = read('system_prompt.txt')
+
+
+DESCRIPTION = '''
+<div>
+<h1 style="text-align: center;">诊疗助手</h1>
+<p>一个帮助您分析症状和检验报告的AI工具。</p>
+<p>🔎 选择您需要咨询的科室，在输入框中输入症状描述或者体检信息等；您也可以在图片框中上传检测报告图。</p>
+<p>🦕 请注意生成信息可能不准确，且不具备任何实际参考价值，如有需要请联系专业医生。</p>
+</div>
+'''
+
+LICENSE = '采用' + MODEL + '模型'
     
 def process_text(text_input, unit):
     if text_input:
         completion = client.chat.completions.create(
             model=MODEL,
             messages=[
-                {"role": "system", "content": f" You are a helpful {unit} doctor." + sys_prompt},
+                {"role": "system", "content": f" You are a experienced {unit} doctor." + SYS_PROMPT},
                 {"role": "user", "content": f"Hello! Could you solve {text_input}?"}
             ]
         )
@@ -47,9 +59,9 @@ def process_image(image_input, unit):
         response = client.chat.completions.create(
             model=MODEL,
             messages=[
-                {"role": "system", "content": f" You are a helpful {unit} doctor." + sys_prompt},
+                {"role": "system", "content": f" You are a experienced {unit} doctor." + SYS_PROMPT},
                 {"role": "user", "content": [
-                    {"type": "text", "text": "Help me understand what this image"},
+                    {"type": "text", "text": "Help me understand what is in this picture and analysis."},
                     {"type": "image_url", 
                      "image_url": {
                         "url": f"data:image/jpeg;base64,{base64_image}",
@@ -72,7 +84,7 @@ def main(text_input="", image_input=None, unit=""):
 
 with gr.Blocks(theme='shivi/calm_seafoam') as iface:
     with gr.Accordion(""):
-        gr.Markdown("""医学报告助手 GPT-4o""")
+        gr.Markdown(DESCRIPTION)
         unit = gr.Dropdown(label="🩺科室", value='内科', elem_id="units",
                             choices=["内科", "外科", "妇产科", "男科", "儿科", \
                                      "五官科", "肿瘤科", "皮肤性病科", "中医科", "传染科", "精神心理科", \
@@ -89,6 +101,8 @@ with gr.Blocks(theme='shivi/calm_seafoam') as iface:
 
     # Set up the event listeners
     submit_btn.click(main, inputs=[text_input, image_input, unit], outputs=output_box)
+
+    gr.Markdown(LICENSE)
     
 #gr.close_all()