Create call_to_api.py

call_to_api.py

@@ -0,0 +1,31 @@
+from openai import OpenAI
+import streamlit as st
+
+def call_openai_api(prompt, model="gpt-4o-mini", api_key=None):
+    """
+    Call the OpenAI API to generate a response based on a given prompt.
+    
+    Parameters:
+        prompt (str): The prompt to generate a response for.
+        model (str): The OpenAI model to use for generating the response.
+        api_key (str): The API key for accessing OpenAI.
+
+    Returns:
+        str: The generated response.   
+    """
+    if api_key is None:
+        st.info("Insira sua chave API OpenAI para continuar.")
+        st.stop()
+
+    openai.api_key = api_key  # Set the API key for OpenAI
+    response = openai.chat.completions.create(
+        model=model,
+        messages=[
+            {"role": "system", "content": "Você é um assistente útil."},
+            {"role": "user", "content": prompt}
+        ],
+        temperature=0.4,
+        max_tokens=1500,
+    )
+    return response.choices[0].message["content"]
+