Spaces:
Sleeping
Sleeping
# -*- coding: utf-8 -*- | |
import requests | |
import streamlit as st | |
import openai | |
from openai import OpenAI | |
import os | |
from dotenv import load_dotenv | |
import numpy as np | |
import pandas as pd | |
import csv | |
import tempfile | |
from tempfile import NamedTemporaryFile | |
import pathlib | |
from pathlib import Path | |
import re | |
from re import sub | |
import matplotlib.pyplot as plt | |
from itertools import product | |
from tqdm import tqdm_notebook, tqdm, trange | |
import time | |
from time import sleep | |
import pretty_errors | |
import seaborn as sns | |
from matplotlib.pyplot import style | |
from rich import print | |
import warnings | |
warnings.filterwarnings('ignore') | |
load_dotenv() | |
### 设置openai的API key | |
os.environ["OPENAI_API_KEY"] = os.environ['user_token'] | |
openai.api_key = os.environ['user_token'] | |
##NOTE: 需要注意OpenAI的版本 | |
def chatgpt(user_prompt, sys_prompt="你是一个咨询顾问,我需要你用简体中文来总结内容。", openai_model="gpt-3.5-turbo-16k"): | |
# print('chatgpt mode starts!') | |
try: | |
openai_client = OpenAI() | |
chatgpt_response = openai_client.chat.completions.create( | |
model=openai_model, | |
messages=[ | |
{"role": "system", "content": sys_prompt}, | |
{"role": "user", "content": user_prompt}, | |
], | |
stream=False, | |
) | |
full_response = chatgpt_response.choices[0].message.content | |
print(full_response) | |
except Exception as e: | |
print(e) | |
full_response = "Sorry, I don't have an answer for that yet." | |
# return chatgpt_response, full_response | |
return full_response | |
# user_prompt = "where is Shanghai?" | |
# system_prompt = "You are a helpful AI assistant." | |
# full_response = chatgpt(user_prompt=user_prompt, sys_prompt=system_prompt) | |
# print(full_response) |