增加使用者可以自己的prompt

app.py

@@ -9,6 +9,7 @@ genaikey=os.environ['genaikey']
 #MODEL_NAME = "seiching/whisper-small-seiching"
 MODEL_NAME = "openai/whisper-tiny"
 BATCH_SIZE = 8
+DEFAULTPROMPT='你是專業的會議紀錄製作員,請根據由語音辨識軟體將會議錄音所轉錄的逐字稿，也請注意逐字稿可能有錯,請先做校正,討論內容細節請略過,請根據校正過的逐字稿撰寫會議紀錄，並要用比較正式及容易閱讀的寫法，避免口語化'
 #
 device = 0 if torch.cuda.is_available() else "cpu"
 
@@ -261,15 +262,15 @@ def gpt4write(openaikeystr,transcribe_text,LLMmodel):
     return finalresponse
 
     return 'ok'
-def gewritenote(inputscript):
+def gewritenote(prompt,inputscript):
     api_key = genaikey
     genai.configure(api_key = api_key)
     model = genai.GenerativeModel('gemini-pro')
-    genprompt='你是專業的會議紀錄製作員,請根據由語音辨識軟體將會議錄音所轉錄的逐字稿，也請注意逐字稿可能有錯,請先做校正,討論內容細節請略過,請根據校正過的逐字稿撰寫會議紀錄，並要用比較正式及容易閱讀的寫法，避免口語化'
-    genprompt=genprompt+'#'+inputscript+'#'
+    #genprompt='你是專業的會議紀錄製作員,請根據由語音辨識軟體將會議錄音所轉錄的逐字稿，也請注意逐字稿可能有錯,請先做校正,討論內容細節請略過,請根據校正過的逐字稿撰寫會議紀錄，並要用比較正式及容易閱讀的寫法，避免口語化'
+    genprompt=prompt+'#'+inputscript+'#'
     response = model.generate_content( genprompt)
     return response.text
-def writenotes( LLMmodel,apikeystr,inputscript):
+def writenotes( LLMmodel,apikeystr,prompt,inputscript):
   #text=transcribe_text
   #openaikey.set_key(inputkey)
   #openaikey = OpenAIKeyClass(inputkey)
@@ -282,7 +283,7 @@ def writenotes( LLMmodel,apikeystr,inputscript):
   elif LLMmodel=="gpt-4-0125-preview": 
       ainotestext=gpt4write(apikeystr,transcribe_text,LLMmodel)
   elif LLMmodel=='gemini':
-      ainotestext=gewritenote(transcribe_text)
+      ainotestext=gewritenote(prompt,transcribe_text)
 
 
 
@@ -298,7 +299,7 @@ def writenotes( LLMmodel,apikeystr,inputscript):
   return ainotestext
 ainotes = gr.Interface(
     fn=writenotes,
-    inputs=[ gr.inputs.Radio(["gemini","gpt-3.5-turbo", "gpt-4-0125-preview"], label="LLMmodel", default="gemini"),gr.Textbox(label="使用GPT請輸入OPEN AI API KEY",placeholder="請輸入sk..."),gr.Textbox(label="逐字稿",placeholder="若沒有做語音辨識，請輸入逐字稿")],
+    inputs=[ gr.inputs.Radio(["gemini","gpt-3.5-turbo", "gpt-4-0125-preview"], label="LLMmodel", default="gemini"),gr.Textbox(label="使用GPT請輸入OPEN AI API KEY",placeholder="請輸入sk..."),gr.Textbox(label="提示詞(prompt)",placeholder=DEFAULTPROMPT),gr.Textbox(label="逐字稿",placeholder="若沒有做語音辨識，請輸入逐字稿")],
     outputs="text",
     layout="horizontal",
     theme="huggingface",