| 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 |