import openai
import os

from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv()) # read local .env file

def get_completion(prompt, model="gpt-3.5-turbo", temperature=0):
    messages = [{"role": "user", "content": prompt}]
    response = openai.ChatCompletion.create(
        model=model,
        messages=messages,
        temperature=temperature, # this is the degree of randomness
    )
    return response.choices[0].message["content"]

prompt = f"""
Translate the following English text to Spanish: \
```Hi, I would like to order a blender```
"""
response = get_completion(prompt)
print(response)

prompt = f"""
Tell me which language this is: 
```Combien coûte le lampadaire?```
"""
response = get_completion(prompt)
print(response)

prompt = f"""
Translate the following text to French and Spanish
and English pirate: 
```I want to order a basketball```
"""
response = get_completion(prompt)
print(response)

prompt = f"""
Translate the following text to Spanish in both the \
formal and informal forms: 
'Would you like to order a pillow?'
"""
response = get_completion(prompt)
print(response)

user_messages = [
    "La performance du système est plus lente que d'habitude.",
    "Mi monitor tiene píxeles que no se iluminan.",
    "Il mio mouse non funziona",
    "Mój klawisz Ctrl jest zepsuty",
    "我的屏幕在闪烁"
]

for issue in user_messages:
    prompt = f"Tell me what language this is: ```{issue}```"
    lang = get_completion(prompt)
    print(f"Original message ({lang}): {issue}")

    prompt = f"""
Translate the following text to English \
and Korean: ```{issue}```
"""
    response = get_completion(prompt)
    print(response, "\n")

prompt = f"""
Translate the following from slang to a business letter: 
'Dude, This is Joe, check out this spec on this standing lamp'
"""
response = get_completion(prompt)
print(response)

data_json = {
    "restaurant_employees":[
        {"name":"Shyam", "email":"shyamjaiswal@gmail.com"},
        {"name":"Bob", "email":"bob32@gmail.com"},
        {"name":"Jai", "email":"jai87@gmail.com"}
    ]
}

prompt = f"""
Translate the following python dictionary from JSON to an HTML \
table with column headers and title: {data_json}
"""
response = get_completion(prompt)
print(response)

from IPython.display import display, Markdown, Latex, HTML

display(HTML(response))

text = [
    "The girl with the black and white puppies have a ball.",
    "Yolanda has her notebook.", # ok
    "Its going to be a long day. Does the car need it's oil changed?",
    "Their goes my freedom. There going to bring they're suitcases.",
    "Your going to need you're notebook.", # Homonyms
    "That medicine effects my ability to sleep. Have you heard of the affect?",
    "This phrase is to cherck chatGPT for speling abilitty" # spelling test
]
for t in text:
    prompt = f"Proofread and correct: ```{t}```"

    prompt = f"""Proofread and correct the following text and rewrite the corrected         version. If you don't find errors, just say "No errors found":
```{t}```
"""
    response = get_completion(prompt)
    print(response)

text = """
Got this for my daughter for her birthday cuz she keeps taking mine from my room. Yes, adults also like pandas too. She takes it everywhere with her, and it's super soft and cute. One ear was a lot lower than the other, and I don't think that were designed to be asymmetrical. It's a bit small for what I paid though. I think there might be other options that are bigger the same price. It arrived a day earlier than expected, so to play with it myself before I gave it to my daughter.
"""

prompt = f"proofread and correct this review: ```{text}```"
response = get_completion(prompt)
print(response)

```python
from redlines import Redlines

diff = Redlines(text,response)
display(Markdown(diff.output_markdown))
```

prompt = f"""
proofread and correct this review. Make it more compelling.
Ensure it follows APA style guide and targets an advanced reader.
Output in markdown format.
Text: ```{text}```
"""

response = get_completion(prompt)
display(Markdown(response))