KDnuggets_keywords_1.py

def keyword_extraction(review):
    response = openai.completions.create(
        model="text-davinci-003",  # You can choose the model version
        prompt=f"""Analyze the following review: \"{review}\". 
        Identify the product that the user is reviewing and answer the product name with one or two words""",
        max_tokens=60
    )

    sentiment = response.choices[0].text.strip()
    return sentiment
  
df_short["Product"] = df_short.apply(lambda x:keyword_extraction(x["Text"]), axis=1)
df_short