Update app.py

app.py

@@ -18,11 +18,6 @@ OCR_TR_DESCRIPTION = '''# OCR Translate v0.2
 # Getting the list of available languages for Tesseract
 choices = os.popen('tesseract --list-langs').read().split('\n')[1:-1]
 
-# Replace YOUR_API_KEY with your API key from https://cloud.google.com/docs/authentication/api-keys
-genai.configure(api_key="AIzaSyC_o3nrbe2fVUifhUg0Zh2oVsweSIdlEbg")
-model = genai.GenerativeModel('gemini-pro')
-
-
 # tesseract语言列表转pytesseract语言
 def ocr_lang(lang_list):
     lang_str = ""
@@ -63,13 +58,13 @@ def cp_clear():
     pyperclip.clear()
 
 # Add a translation function
-def translate(input_text, inputs_transStyle):
+def translate(api_key, input_text, inputs_transStyle):
+    genai.configure(api_key=api_key)
+    model = genai.GenerativeModel('gemini-pro')
     if input_text is None or input_text == "":
         return "System prompt: There is no content to translate!"
         
-    # 选择翻译模型
-    prompt = f"Please reformat the following article to have clear paragraph breaks and correct punctuation, and then translate it into {inputs_transStyle}. In the translation, do not display the original text, fictional content, or any repeated content. Ensure that the original meaning and context are preserved as much as possible."
-    # Reorder for ease of reading and translate into {inputs_transStyle}"
+    prompt = f"Please reformat the following article to have clear paragraph breaks and correct punctuation. Then, translate it into {inputs_transStyle}, ensuring that the original meaning and context are preserved as much as possible. In the translation, ensure that no content is repeated."
     response = model.generate_content([prompt, input_text])
     return response.text
 
@@ -98,15 +93,19 @@ def main():
                         clear_img_btn = gr.Button('Clear')
                         ocr_btn = gr.Button(value='OCR Extraction', variant="primary")
 
+                    with gr.Row():
+                        inputs_api_key = gr.inputs.Textbox(label='API: https://makersuite.google.com/u/1/app/apikey', type='password')
+
                 with gr.Column():
                     with gr.Row():
                         outputs_text = gr.Textbox(label="Extract content", lines=20)
                     inputs_transStyle = gr.inputs.Dropdown(choices=["Chinese (Simplified)", "Chinese (Traditional)", "English", "Japanese", "Korean"], 
-                                                           default="Chinese (Simplified)", label='translation mode')
+                                                           default="Chinese (Traditional)", label='translation mode')
                     with gr.Row():
                         clear_text_btn = gr.Button('Clear')
                         translate_btn = gr.Button(value='Translate', variant="primary")
 
+
             with gr.Row():
                 pass
 
@@ -129,14 +128,14 @@ def main():
         clear_img_btn.click(fn=clear_content, inputs=[], outputs=[inputs_img])
 
         # ---------------------- 翻译 ----------------------
-        translate_btn.click(fn=translate, inputs=[outputs_text, inputs_transStyle], outputs=[outputs_tr_text])
+        translate_btn.click(fn=translate, inputs=[inputs_api_key, outputs_text, inputs_transStyle], outputs=[outputs_tr_text])
         clear_text_btn.click(fn=clear_content, inputs=[], outputs=[outputs_text])
 
         # ---------------------- 复制到剪贴板 ----------------------
         cp_btn.click(fn=cp_text, inputs=[outputs_tr_text], outputs=[])
         cp_clear_btn.click(fn=cp_clear, inputs=[], outputs=[])
 
-    ocr_tr.launch(inbrowser=True)
+    ocr_tr.launch(inbrowser=True, share=True)
 
 if __name__ == '__main__':
     main()