Project 3 AI Research Assistant

pip install pandas openai

from openai import OpenAI
import pandas as pd

client = OpenAI(api_key="YOUR_API_KEY")

df = pd.read_csv("/path/to/your/user_behavior_dataset.csv")

def analyze_data(df):
    csv_content = df.to_csv(index=False)
    
    response = client.chat.completions.create(
        model="gpt-4",
        messages=[
            {
                "role": "user",
                "content": (
                    "You are a research assistant. "
                    "Provide key insights from the following dataset in bullet points:\n"
                    f"{csv_content}"
                )
            }
        ],
        max_tokens=500,
        temperature=0.2
    )
    return response.choices[0].message.content

if __name__ == "__main__":
    summary = analyze_data(df)
    print(summary)

def analyze_demographics(df):
    demo_df = df[['age', 'gender', 'country']]
    csv_content = demo_df.to_csv(index=False)
    
    response = client.chat.completions.create(
        model="gpt-4",
        messages=[
            {
                "role": "user",
                "content": (
                    "You are a research assistant. "
                    "Provide key demographic insights from the following data in bullet points:\n"
                    f"{csv_content}"
                )
            }
        ],
        max_tokens=300,
        temperature=0.2
    )
    return response.choices[0].message.content

if __name__ == "__main__":
    demographics_summary = analyze_demographics(df)
    print(demographics_summary)