Demo Performing Text Processing and Analysis

# In [1]
import os
import openai

openai.api_key = os.getenv("OPENAI_API_KEY")
def get_word_completion(prompt: str, model: str = "gpt-3.5-turbo") -> str:
    """
    Send a chat-style prompt and return the assistant's content string.
    """
    messages = [
        {"role": "system", "content": "You are a helpful assistant."},
        {"role": "user", "content": prompt},
    ]
    response = openai.ChatCompletion.create(model=model, messages=messages)
    return response.choices[0].message.content

# In [2]: Context (excerpt)
context = '''
Steve Jobs' 2005 Stanford Commencement Address
I am honored to be with you today at your commencement from one of the finest universities in the world. I never gradu

The first story is about connecting the dots.

I dropped out of Reed College after the first 6 months, but then stayed around as a drop-in for another 18 months or so
'''

# In [3]: Prompt asking for a 500-word summary
prompt = f"""
Create a summary capturing the main points and key details in 500 words based on the content delimited by triple backticks.

```{context}```
"""
response = get_word_completion(prompt)
print(response)

# In [4]: Bullet summary prompt
prompt_bullets = f"""
Create a summary capturing the main points and key details as bullets based on the content delimited by triple backticks.

```{context}```
"""
response_bullets = get_word_completion(prompt_bullets)
print(response_bullets)

# In [5]: Sentiment analysis context
context_reviews = '''
1. If you sometimes like to go to the movies to have fun, Wasabi is a good place to start.
2. An idealistic love story that brings out the latent 15-year-old romantic in everyone.
3. The story loses its bite in a last-minute happy ending that's even less plausible than the rest of the picture.
'''
prompt_sentiment = f"""
Analyze the sentiment of the reviews delimited in triple backticks.

First show the actual review and then add the sentiment - Positive, Negative, or Neutral.

```{context_reviews}```
"""
response_sentiment = get_word_completion(prompt_sentiment)
print(response_sentiment)

# In [6]: Poem translation context
context_poem = """
Demain, dès l'aube, à l'heure où blanchit la campagne,
Je partirai. Vois-tu, je sais que tu m'attends.
J'irai par la forêt, j'irai par la montagne.
Je ne puis demeurer loin de toi plus longtemps.
Je marcherai les yeux fixés sur mes pensées,
Sans rien voir au dehors, sans entendre aucun bruit,
Seul, inconnu, le dos courbé, les mains croisées,
Triste, et le jour pour moi sera comme la nuit.
Je ne regarderai ni l'or du soir qui tombe,
Ni les voiles au loin descendant vers Harfleur,
Et quand j'arriverai, je mettrai sur ta tombe
Un bouquet de houx vert et de bruyère en fleur.
"""
prompt_translate = f"""
Write an English poem based on the French poem delimited in triple backticks.

```{context_poem}```
"""
response_translate = get_word_completion(prompt_translate)
print(response_translate)

# In [7]: States and capitals context
context_states = """
1. Alabama - Montgomery
2. California - Sacramento
3. Florida - Tallahassee
4. Georgia - Atlanta
5. Illinois - Springfield
6. Massachusetts - Boston
7. New York - Albany
8. Texas - Austin
9. Pennsylvania - Harrisburg
10. Washington - Olympia
"""
prompt_formats = f"""
From the content delimited in triple backticks, format it in JSON, XML, and JSONL.
```{context_states}```
"""
response_formats = get_word_completion(prompt_formats)
print(response_formats)

[
  {
    "state": "Alabama",
    "capital": "Montgomery"
  },
  {
    "state": "California",
    "capital": "Sacramento"
  },
  ...
]

<root>
  <state>
    <name>Alabama</name>
    <capital>Montgomery</capital>
  </state>
  <state>
    <name>California</name>
    <capital>Sacramento</capital>
  </state>
  ...
</root>

{"state": "Alabama", "capital": "Montgomery"}
{"state": "California", "capital": "Sacramento"}
{"state": "Florida", "capital": "Tallahassee"}
{"state": "Georgia", "capital": "Atlanta"}
{"state": "Illinois", "capital": "Springfield"}
{"state": "Massachusetts", "capital": "Boston"}
{"state": "New York", "capital": "Albany"}
{"state": "Texas", "capital": "Austin"}
{"state": "Pennsylvania", "capital": "Harrisburg"}
{"state": "Washington", "capital": "Olympia"}