Converting the interface to summarization

app.py

@@ -2,7 +2,8 @@ import os
 import gradio as gr
 from dotenv import load_dotenv
 import openai
-from utils import serialize
+
+# from utils import serialize
 from utils import compress
 
 from description import DESCRIPTION
@@ -14,19 +15,42 @@ OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
 openai.api_key = OPENAI_API_KEY
 
 
-def chat(message, history):
-    """
-    Sends a request to the OpenAi api based on the user input and the history
-    """
-    messages = serialize(history)
-    messages.append({"role": "user", "content": message})
+def load_prompt(path):
+    with open(path) as f:
+        lines = f.readlines()
+        return "".join(lines)
+
+
+def chat(passage, max_tokens=256, temprature=0, debug=False):
+
+    if debug:
+        passage = """
+        A car or automobile is a motor vehicle with wheels. Most definitions of cars say that they run primarily on roads, seat one to eight people, have four wheels, and mainly transport people (rather than goods).
+        """
+
+    prompt = load_prompt("summary_prompt.txt").replace("<<SUMMARY>>", passage)
 
-    completion = openai.ChatCompletion.create(
+    summary = openai.ChatCompletion.create(
         model="gpt-3.5-turbo",
-        messages=messages,
+        messages=[{"role": "user", "content": prompt}],
     )
 
-    return completion["choices"][0]["message"]["content"].strip()
+    return summary["choices"][0]["message"]["content"].strip()
+
+
+# def chat(message, history):
+#     """
+#     Sends a request to the OpenAi api based on the user input and the history
+#     """
+#     messages = serialize(history)
+#     messages.append({"role": "user", "content": message})
+
+#     completion = openai.ChatCompletion.create(
+#         model="gpt-3.5-turbo",
+#         messages=messages,
+#     )
+
+#     return completion["choices"][0]["message"]["content"].strip()
 
 
 def transcribe(audio_file):
@@ -40,7 +64,7 @@ def predict(input, history=[]):
     compress(input)
     transcription = transcribe(input)
 
-    answer = chat(transcription, history)
+    answer = chat(transcription)
     history.append((transcription, answer))
     response = history
     return response, history

summary_prompt.txt

@@ -0,0 +1,5 @@
+Write a detailed summary of the following:
+
+<<SUMMARY>>
+
+DETAILED SUMMARY:

utils.py

@@ -3,22 +3,22 @@ from pathlib import Path, PurePath
 import soundfile as sf
 
 
-def serialize(messages):
-    """
-    Converts a list of tuples where each element of the list represents a message to dictionary of
-    messages
-    """
+# def serialize(messages):
+#     """
+#     Converts a list of tuples where each element of the list represents a message to dictionary of
+#     messages
+#     """
 
-    serialized_messages = []
-    for message in messages:
-        serialized_message = [
-            {"role": "system", "content": message[0]},
-            {"role": "user", "content": message[1]},
-        ]
+#     serialized_messages = []
+#     for message in messages:
+#         serialized_message = [
+#             {"role": "system", "content": message[0]},
+#             {"role": "user", "content": message[1]},
+#         ]
 
-        serialized_messages.extend(serialized_message)
+#         serialized_messages.extend(serialized_message)
 
-    return serialized_messages
+#     return serialized_messages
 
 
 def compress(audio_file):