| from openai import OpenAI |
| import os |
|
|
| client = OpenAI(api_key=os.environ["OPENAI_API_KEY"]) |
|
|
| def generate_scenario(gender): |
| PROMPT = f"""Create a scenario where you're see a {gender} you're interested in. You haven't yet approached them yet but are interested. Keep it brief, around 20 words. For example: 'There's a cute {gender} in my boxing class. I approach her after a class.' or 'I see two {gender}s having coffee at the table next to mine.'.""" |
| return llm(PROMPT) |
|
|
| def suggest_next_line(scenario, history): |
| PROMPT = f"""This is my scenario: {scenario} \n\n""" |
| if len(history) == 0: |
| PROMPT += "Suggest a line with which I can introduce myself or strike up a conversation." |
| else: |
| PROMPT += f"""This is the conversation so far: \n\n{format_history(history)} \n\nSuggest my next response. Provide the response and nothing else.""" |
| return llm(PROMPT) |
|
|
| def generate_response(scenario, history, personality, interested): |
| PROMPT = f"""This is my scenario: {scenario}. Roleplay the conversation with me. Provide the response and nothing else. You're personality is {personality}. You are {'interested' if interested else 'not interested'} in me. |
| |
| The conversation so far: \n\n{format_history(history)} |
| |
| Only provide the response and nothing else. |
| """ |
| return llm(PROMPT) |
|
|
| def transcribe_audio(audio_path): |
| with open(audio_path, "rb") as audio_file: |
| transcript = client.audio.transcriptions.create( |
| model="whisper-1", |
| file=audio_file |
| ) |
| return transcript.text |
|
|
| def llm(prompt): |
| response = client.completions.create( |
| model="gpt-3.5-turbo-instruct", |
| prompt=prompt, |
| temperature=0.9, |
| max_tokens=100, |
| top_p=1, |
| frequency_penalty=0.0, |
| presence_penalty=0.6, |
| ) |
| return response.choices[0].text.strip() |
| |
|
|
| def format_history(history): |
| return "\n\n".join([f"Me: {line[0]} \nThem: {line[1] if line[1] else ''}" for line in history]) |
|
|
