import os
import streamlit as st
import requests

def get_completion_from_openai(prompt, max_tokens = None):
    url = os.getenv('OPENAI_COMPLETION_URL')

    headers = {
        "Content-Type": "application/json",
        "Authorization": "Bearer " + st.secrets["OPENAI_TOKEN"],
    }

    response = requests.post(url,
                                 json={    
                                     "model": os.getenv('OPENAI_MODEL'),
                                     "max_tokens": max_tokens,
                                     "messages": [
                                      {
                                        "role": "user",
                                        "content": prompt
                                      }
                                    ]
                                }, 
                                 headers=headers, 
                                 stream=False,
                                 )
    try:
        return response.json()['choices'][0]['message']['content']
    except:
        print(response.json())
        return "Произошла ошибка"