Python string column iteration

Question

I am working on openAI, and stuck I have tried to sort this issue on my own but didn't get any resolution. I want my code to run the sentence generation operation on every row of the Input_Description_OAI column and give me the output in another column (OpenAI_Description). Can someone please help me with the completion of this task. I am new to python.

The dataset looks like:

    import os
    import openai
    import wandb
    import pandas as pd
    openai.api_key = "MY-API-Key"
    data=pd.read_excel("/content/OpenAI description.xlsx")
    data
        
    data["OpenAI_Description"] = data.apply(lambda _: ' ', axis=1)
    data
        
    gpt_prompt = ("Write product description for: Brand: COILCRAFT ; MPN: DO5010H-103MLD..")
        
        
    response = openai.Completion.create(engine="text-curie-001", prompt=gpt_prompt, 
    temperature=0.7, max_tokens=1000, top_p=1.0, frequency_penalty=0.0, presence_penalty=0.0)
        
        
    print(response['choices'][0]['text'])
    data['OpenAI_Description'] = data.apply(gpt_prompt,response['choices'][0]['text'], axis=1)

I got the error after execution on first row as:

---------------------------------------------------------------------------
TypeError 
Traceback (most recent call last)
<ipython-input-32-c798fbf9bc16> in <module>
     15 print(response['choices'][0]['text'])
     16 #data.add_data(gpt_prompt,response['choices'][0]['text'])
---> 17 data['OpenAI_Description'] = data.apply(gpt_prompt,response['choices'][0]['text'], axis=1)
     18 

TypeError: apply() got multiple values for argument 'axis'

Python string column iteration

0 Answers0