Update app.py

app.py

@@ -57,16 +57,50 @@ def cp_text(input_text):
 def cp_clear():
     pyperclip.clear()
 
+# Split the text into 4000 character chunks
+def process_text_input_text(input_text):
+    # Split the text into 4000 character chunks
+    chunks = [input_text[i:i+4000] for i in range(0, len(input_text), 4000)]
+    return chunks
+
+def process_and_translate(api_key, input_text, inputs_transStyle):
+    # Process the input text into chunks
+    chunks = process_text_input_text(input_text)
+
+    # Translate each chunk and collect the results
+    translated_chunks = []
+    for chunk in chunks:
+        if chunk is None or chunk == "":
+            translated_chunks.append("System prompt: There is no content to translate!")
+        else:
+            prompt = f"Display language is {inputs_transStyle}, do not display original text, As a Knowledge Video Content Analysis Expert, I specialize in analyzing knowledge videos, identifying and clearly explaining key points in {inputs_transStyle}, ensuring accurate, easy-to-understand summaries suitable for diverse audiences, analyze, list key points, and explain detailedly below text: "
+            genai.configure(api_key=api_key)
+            model = genai.GenerativeModel('gemini-pro')
+            response = model.generate_content([prompt, chunk],
+                        generation_config=genai.types.GenerationConfig(
+                            # Only one candidate for now.
+                            candidate_count=1,
+                            stop_sequences=['ʤ'],
+                            max_output_tokens=2048,
+                            temperature=1.0)
+                    )
+            translated_chunks.append(response.text)
+
+    # Join the translated chunks back together into a single string
+    response = '\n--------------------------------------------------\n'.join(translated_chunks)
+
+    return response
+
 # Add a translation function
-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"In the translation, ensure that no content is repeated and original meaning and context are preserved as much as possible. 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."
-    response = model.generate_content([prompt, input_text])
-    return response.text
+# 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"Via {inputs_transStyle}, As a Knowledge Video Content Analysis Expert, I specialize in analyzing knowledge videos, identifying and clearly explaining key points in English, ensuring accurate, easy-to-understand summaries suitable for diverse audiences."
+#     response = model.generate_content([prompt, input_text])
+#     return response.text
 
 def main():
 
@@ -100,7 +134,7 @@ def main():
                     with gr.Row():
                         # Create a text input box for users to enter their API key
                         inputs_api_key = gr.Textbox(label="Please enter your API key here", type="password")
-                        
+
                 with gr.Column():
                     with gr.Row():
                         outputs_text = gr.Textbox(label="Extract content", lines=20)
@@ -133,7 +167,7 @@ def main():
         clear_img_btn.click(fn=clear_content, inputs=[], outputs=[inputs_img])
 
         # ---------------------- 翻译 ----------------------
-        translate_btn.click(fn=translate, inputs=[inputs_api_key, outputs_text, inputs_transStyle], outputs=[outputs_tr_text])
+        translate_btn.click(fn=process_and_translate, inputs=[inputs_api_key, outputs_text, inputs_transStyle], outputs=[outputs_tr_text])
         clear_text_btn.click(fn=clear_content, inputs=[], outputs=[outputs_text])
 
         # ---------------------- 复制到剪贴板 ----------------------