Main working

interviewkit/interview.py

@@ -1,7 +1,7 @@
 from random import choice
 from interviewee import Interviewee
-import openai
 from enum import StrEnum, auto
+from transcript import Transcript
 
 
 class Status(StrEnum):
@@ -14,7 +14,7 @@ class Status(StrEnum):
 class Interview(object):
     """An interview between an Interviewee and an Interviewer."""
 
-    def __init__(self, interviewee: Interviewee):
+    def __init__(self, interviewee: Interviewee, transcript: Transcript):
         self.interviewee = interviewee
         self.status = Status.IN_PROGRESS
 
@@ -27,8 +27,8 @@ def stop_recording(self) -> None:
         self.status = Status.RECORDED
 
     def start_transcription(self) -> None:
+        print("transcribing")
         self.status = "Transcribing"
-        self.transcript = []  # TODO: Use object Transcript instead of list?
 
     def add_to_transcript(self, speech: str) -> None:
         self.transcript.append(speech)
@@ -40,20 +40,15 @@ def stop_transcription(self) -> None:
         self.transcript = " ".join(self.transcript)
 
     def suggest_questions(self) -> str:
-        """Use the OpenAI API to suggest questions based on the transcript content."""
+        """suggest questions based on the transcript content."""
         suggested_questions = self.generate_questions(self.transcript)
         return suggested_questions
 
     def generate_questions(self, content: str) -> list[str]:
-        """Call the OpenAI API to generate questions, and return them."""
-        response = openai.Completion.create(
-            engine="text-davinci-002",
-            prompt=f"Generate interview questions based on the following transcript:\n{content}\n",
-            max_tokens=50,  # Adjust the max tokens as needed
-            n=5,  # Number of questions to generate
-            stop=None,
-            temperature=0.7,  # Adjust the temperature for creativity
-        )
+        """generate questions, and return them."""
+
+        #TODO
+        response = None
 
         # Extract and return the generated questions
         generated_questions = [choice["text"].strip() for choice in response.choices]

interviewkit/interviewee.py

@@ -1,7 +1,6 @@
 from enum import StrEnum, auto
 from pydantic import BaseModel, NonNegativeInt
 
-
 class Gender(StrEnum):
     MALE = auto()
     FEMALE = auto()
@@ -13,4 +12,4 @@ class Interviewee(BaseModel):
 
     name: str
     age: NonNegativeInt
-    gender: Gender
+    gender: Gender

interviewkit/main.py

@@ -1,26 +1,20 @@
 import os
 
-import openai
-
 from interview import Interview
-from interviewee import Interviewee
+from interviewee import Interviewee, Gender
 from transcript import Transcript
 from settings import Settings
 
-
-# Set the OpenAI API key
 settings = Settings()
-openai.api_key = settings.OPENAI_API_KEY
-
 
 if __name__ == "__main__":
-    interviewee = Interviewee("John Doe", 60, "Male")
+    interviewee = Interviewee(name="John Doe", age=60, gender=Gender.MALE)
 
     transcript_content = "This is the content of the transcript."
-    transcript = Transcript(transcript_content)
+    transcript = Transcript(content=transcript_content)
 
     interview = Interview(interviewee, transcript)
 
-    interview.transcribe()
+    interview.start_transcription()
 
     print(f"Interview Status: {interview.status}")

interviewkit/settings.py

@@ -2,4 +2,5 @@
 
 
 class Settings(BaseSettings):
-    OPENAI_API_KEY: str
+    #TODO
+    pass

pyproject.toml

@@ -8,7 +8,6 @@ license= "GPL-2.0-or-later"
 
 [tool.poetry.dependencies]
 python = "^3.11"
-openai = "^0.28.0"
 pydantic-settings = "^2.0.3"
 pydantic = "^2.3.0"
 

requirements.txt

@@ -8,7 +8,6 @@ colorama==0.4.6
 frozenlist==1.4.0
 idna==3.4
 multidict==6.0.4
-openai==0.28.0
 requests==2.31.0
 tqdm==4.66.1
 urllib3==2.0.5