update genmin

app.py

@@ -5,6 +5,7 @@ import gradio as gr
 import os
 
 hugapikey=os.environ['openaikey']
+genaikey=os.environ['genaikey']
 MODEL_NAME = "seiching/whisper-small-seiching"
 #MODEL_NAME = "openai/whisper-small"
 BATCH_SIZE = 8
@@ -89,15 +90,17 @@ def process_chunks(openaikeystr,inputtext):
     #openaikey.set_key(openaikeystr)
     #print('process_chunk',openaikey.get_key())
     chunks = split_into_chunks(text)
-    response=''
+    response='這是分段會議紀錄結果\n\n'
     i=1
     for chunk in chunks:
       
       response=response+'第' +str(i)+'段\n'+call_openai_api(openaiobj,chunk)+'\n\n'
       i=i+1
       # response=response+call_openai_summary(openaiobj,chunk)
-
-    finalresponse=response+'\n\n summary \n\n' +call_openai_api(openaiobj,response)
+    if i>2:
+        finalresponse=response+'\n\n 這是根據以上分段會議紀錄彙編如下 \n\n' +call_openai_api(openaiobj,response)
+    else:
+        finalresponse=response
     return finalresponse
     # # Processes chunks in parallel
     # with ThreadPoolExecutor() as executor:
@@ -208,15 +211,32 @@ file_transcribe = gr.Interface(
     cache_examples=True,
     allow_flagging="never",
 )
-def writenotes( apikeystr,inputscript):
+import google.generativeai as genai
+def gewritenote(inputscript):
+    api_key = gekey
+    genai.configure(api_key = api_key)
+    model = genai.GenerativeModel('gemini-pro')
+    genprompt='你是專業的會議紀錄製作員,請根據由語音辨識軟體將會議錄音所轉錄的逐字稿，也請注意逐字稿可能有錯,討論內容細節請略過,請列出經主席確認的會議決議，並要用比較正式及容易閱讀的寫法，避免口語化'
+    genprompt=genprompt+'#'+inputscript+'#'
+    response = model.generate_content( genprompt)
+    return response.text
+def writenotes( LLMmodel,apikeystr,inputscript):
   #text=transcribe_text
   #openaikey.set_key(inputkey)
   #openaikey = OpenAIKeyClass(inputkey)
   print('ok')
-  if len(inputscript)>10:
-    transcribe_text=inputscript
-   
-  ainotestext=process_chunks(apikeystr,transcribe_text)
+  if len(inputscript)>10: #有資料表示不是來自語音辨識結果
+      transcribe_text=inputscript
+  if LLMmodel=="gpt-3.5-turbo": 
+      ainotestext=process_chunks(apikeystr,transcribe_text)
+  elif LLMmodel=="gpt-4-0125-preview": 
+      ainotestext=gpt4write(apikeystr,transcribe_text)
+  elif LLMmodel=='gemini':
+      ainotestext=gewritenote(inputscript)
+
+
+
+
    # ainotestext=inputscript
   #ainotestext=""
   # with open('asr_resul.txt', 'w') as f:
@@ -228,7 +248,7 @@ def writenotes( apikeystr,inputscript):
   return ainotestext
 ainotes = gr.Interface(
     fn=writenotes,
-    inputs=[gr.Textbox(label="OPEN AI API KEY",placeholder="請輸入sk..."),gr.Textbox(label="逐字稿",placeholder="若沒有做語音辨識，請輸入逐字稿")],
+    inputs=[ gr.inputs.Radio(["gpt-3.5-turbo", "gpt-4-0125-preview","gemini"], label="LLMmodel", default="gpt-3.5-turbo"),gr.Textbox(label="使用GPT請輸入OPEN AI API KEY",placeholder="請輸入sk..."),gr.Textbox(label="逐字稿",placeholder="若沒有做語音辨識，請輸入逐字稿")],
     outputs="text",
     layout="horizontal",
     theme="huggingface",