aa

app.py

@@ -11,6 +11,9 @@ def init_state(state):
 
     if (len(state["summarize_history"]) == 0):
         state["summarize_history"] = []
+    
+    if (len(state["title_history"]) == 0):
+        state["title_history"] = ""
 
 def make_return_message(state):
     init_state(state)
@@ -18,11 +21,14 @@ def make_return_message(state):
     if len(state["summarize_history"]) > 0:
         message_string = '\n'.join(state["message_history"])
         summarize_string = '\n'.join(state["summarize_history"])
+        title = state["title_history"]
         return f"""
-# 要約
+# {title}
+
+## 要約
 {summarize_string}
 
-# 本文
+## 本文
 {message_string}
 """
     else:
@@ -32,12 +38,13 @@ def make_return_message(state):
 """
 
 def summarize_text(state):
-    def modify_text():
+        
+    def modify_text(state):
         message_string = '\n'.join(state["message_history"])
 
         prompt = f"""
     # 命令
-    以下の文章の誤字脱字を修正してください。
+    以下の文章の誤字脱字を修正して、適切に段落をつけてください。
     出力は修正後の文章だけにしてください。
 
     # 修正対象の文章
@@ -48,7 +55,7 @@ def summarize_text(state):
         response =  openai.ChatCompletion.create(
             # model="gpt-3.5-turbo-0613", 
             model="gpt-3.5-turbo-16k",
-            max_tokens=1000,
+            # max_tokens=1000,
             messages=[
                 {"role": "user", "content": prompt}
                 ]
@@ -56,9 +63,37 @@ def summarize_text(state):
         assistant_msg = response["choices"][0]["message"]["content"]
         print(assistant_msg)
         state["message_history"] = assistant_msg.split('\n')
-        
+    
+    def make_title(state):
+        message_string = '\n'.join(state["message_history"])
+
+        prompt = f"""
+    # 命令
+    以下の文章に、全体の内容を分かりやすく表現する適切なタイトル文をつけてください。
+    タイトル文は20文字以内にしてください。
+    出力はタイトル文だけにしてください。
+
+    # 文章
+    {message_string}
+
+    # Output
+    """
+        response =  openai.ChatCompletion.create(
+            # model="gpt-3.5-turbo-0613", 
+            model="gpt-3.5-turbo-16k",
+            # max_tokens=1000,
+            messages=[
+                {"role": "user", "content": prompt}
+                ]
+        )
+        assistant_msg = response["choices"][0]["message"]["content"]
+        print(assistant_msg)
+        state["title_history"] = assistant_msg
+        print(state["title_history"])
+
     init_state(state)
-    modify_text()
+    modify_text(state)
+    make_title(state)
 
     message_string = '\n'.join(state["message_history"])
 
@@ -75,7 +110,7 @@ def summarize_text(state):
     response =  openai.ChatCompletion.create(
         # model="gpt-3.5-turbo-0613",  
         model="gpt-3.5-turbo-16k",
-        max_tokens=1000,
+        # max_tokens=1000,
         messages=[
             {"role": "user", "content": prompt}
             ]
@@ -116,6 +151,9 @@ def transcribe(audio,state):
 
     return make_return_message(state)
 
+
+
+
 user_db = {"a": "aa", "yourmystar": "amazing"}
 def auth(user_name, password):
     return user_db.get(user_name) == password
@@ -123,6 +161,7 @@ def auth(user_name, password):
 with gr.Blocks() as app: 
     state = gr.State({
         "message_history" : [],
+        "title_history" : [],
         "summarize_history" : []
     })
     with gr.Row(): 
@@ -141,7 +180,8 @@ with gr.Blocks() as app:
             5. 「**要約をつける**」を押すと、ChatGPTが誤字脱字を修正し、要約をつけてくれます
 
             ※ブラウザ更新すると文章が消えるので、忘れずにコピペしてください<br>
-            ※クリップボードがうまく動作していません
+            ※クリップボードがうまく動作していません<br>
+            ※要約できるのは1万文字くらいまでです
             """)
 
             inputs = gr.Audio(source="microphone", type="filepath", show_label="true",label="話しかけてください")