utils/summarizer.py

import ast
import openai

# Prompt engineering
def get_prompt(text):
    prompt_prefix = """Generate a markdown format weekly report for me by those keywords: {}""" 
    prompt = prompt_prefix.format(text)
    return prompt

# Chat completion
def get_openai_chatcompletion(keywords):
    """Get OpenAI Chat Completion result.
    """
    messages = []
    processed_text = keywords
    augmented_prompt = get_prompt(processed_text)
    messages.append({"role":"user","content": augmented_prompt})
    
    try:
        result = openai.ChatCompletion.create(
            model="gpt-3.5-turbo",
            messages=messages,
            temperature=0.7
        )
    except:
        raise
    return result

# Process result
def get_analyze(result):
    try:
        analyze = result["choices"][0]["message"]["content"]
    except:
        raise    
    return analyze

def get_analyze_result(keywords):
    result = get_openai_chatcompletion(keywords)
    analyze = get_analyze(result)
    return analyze