Update app.py

app.py

@@ -5,28 +5,23 @@ import os
 import io
 import base64
 
-# Set API key and organization ID from environment variables
-api_key = os.environ.get("OPENAI_API_KEY")
+# Set API key
 #base_url = os.environ.get("OPENAI_API_BASE")
+api_key = ""
 client = OpenAI(api_key=api_key)
 
 # Define the model to be used
 MODEL = os.environ.get("MODEL")
 
-def read(filename):
-    with open(filename) as f:
-        data = f.read()
-    return data
-    
 SYS_PROMPT = read('system_prompt.txt')
 
 
 DESCRIPTION = '''
 <div>
-<h1 style="text-align: center;">诊疗助手Alpha</h1>
-<p>一个帮助您分析症状和检验报告的AI工具。</p>
-<p>🔎 选择您需要咨询的科室，在输入框中输入症状描述或者体检信息等；您也可以在图片框中上传检测报告图。</p>
-<p>🦕 请注意生成信息可能不准确，且不具备任何实际参考价值，如有需要请联系专业医生。</p>
+<h1 style="text-align: center;">Medster - Diagnostic Assistant</h1>
+<p>An AI tool that helps you analyze symptoms and test reports.  </p>
+<p>🔎 Select the department you need to consult, and enter the symptom description or physical examination information in the input box; you can also upload the test report image in the picture box.  </p>
+<p>🦕 Please note that the generated information may be inaccurate and does not have any actual reference value. Please contact a professional doctor if necessary.  </p>
 </div>
 '''
 
@@ -42,7 +37,12 @@ footer {
 """
 
 
-LICENSE = '采用 ' + MODEL + ' 模型'
+LICENSE = 'MODEL: ' + MODEL + ' LOADED'
+
+def read(filename):
+    with open(filename) as f:
+        data = f.read()
+    return data
     
 def process_text(text_input, unit):
     if text_input:
@@ -95,22 +95,24 @@ def main(text_input="", image_input=None, unit=""):
 with gr.Blocks(theme='shivi/calm_seafoam', css=css, title="诊疗助手Alpha") as iface:
     with gr.Accordion(""):
         gr.Markdown(DESCRIPTION)
-        unit = gr.Dropdown(label="🩺科室", value='中医科', elem_id="units",
-                            choices=["中医科", "内科", "外科", "妇产科",  "儿科", \
-                                     "五官科", "男科", "皮肤性病科", "传染科", "精神心理科", \
-                                        "整形美容科", "营养科", "生殖中心", "麻醉医学科", "医学影像科", \
-                                            "骨科", "肿瘤科", "急诊科", "检验科"])
+        unit = gr.Dropdown(label="🩺Department", value='Traditional Chinese Medicine', elem_id="units",
+                             choices=["Traditional Medicine", "Internal Medicine", "Surgery", "Obstetrics and Gynecology", "Pediatrics", \
+                                      "Orthodontics", "Andrology", "Dermatology and Venereology", "Infectious Diseases", "Psychiatry", \
+                                         "Plastic Surgery Department", "Nutrition Department", "Reproductive Center", "Anesthesiology Department", "Medical Imaging Department", \
+                                             "Orthopedics", "Oncology", "Emergency Department", "Laboratory Department"])
+    with gr.Row():
+        output_box = gr.Markdown(label="Diagnosis")  # Create an output textbox
     with gr.Row():
-        output_box = gr.Markdown(label="分析")  # Create an output textbox
+        api_key = gr.Textbox(label="OpenAI API Key") # Input API key
     with gr.Row():
-        image_input = gr.Image(type="pil", label="上传图片")  # Create an image upload button
-        text_input = gr.Textbox(label="输入")  # Create a text input box
+        image_input = gr.Image(type="pil", label="Upload Image")  # Create an image upload button
+        text_input = gr.Textbox(label="Submit")  # Create a text input box
     with gr.Row():
-        submit_btn = gr.Button("🚀 确认")  # Create a submit button
-        clear_btn = gr.ClearButton(output_box, value="🗑️ 清空") # Create a clear button
+        submit_btn = gr.Button("🚀 Send")  # Create a submit button
+        clear_btn = gr.ClearButton(output_box, value="🗑️ Clear") # Create a clear button
 
     # Set up the event listeners
-    submit_btn.click(main, inputs=[text_input, image_input, unit], outputs=output_box)
+    submit_btn.click(main, inputs=[apiKey, text_input, image_input, unit], outputs=output_box)
 
     gr.Markdown(LICENSE)