Science4Insights commited on
Commit
d811bbb
·
verified ·
1 Parent(s): 80ca660

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -13
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  import tempfile
5
  import os
6
  from typing import Dict
7
- from openai import OpenAI
8
  from tqdm.auto import tqdm
9
  import plotly.express as px
10
  import plotly.graph_objects as go
@@ -135,22 +135,16 @@ def procesar_respuesta_api(respuesta):
135
  return responses
136
 
137
  def llamar_api_gpt(descripcion, encuesta, api_key):
138
- client = OpenAI(api_key=api_key)
139
 
140
  system_content = f"Eres una persona que responde a estas características: \n\n{descripcion}\n\nTu misión consiste en responder a una encuesta. Tu respuesta debe tener este formato. Número de pregunta y opción. Por ejemplo, P1. a), P2. b), etcétera.\n\nEs muy importante que respondas poniéndote en el rol que se te ha dado. No añadas espacios ni saltos de página. La encuesta es: "
141
 
142
  try:
143
- response = client.chat.completions.create(
144
- model="gpt-4",
145
  messages=[
146
- {
147
- "role": "system",
148
- "content": system_content
149
- },
150
- {
151
- "role": "user",
152
- "content": encuesta
153
- }
154
  ],
155
  temperature=0,
156
  max_tokens=200,
@@ -158,7 +152,7 @@ def llamar_api_gpt(descripcion, encuesta, api_key):
158
  frequency_penalty=0,
159
  presence_penalty=0
160
  )
161
- return response.choices[0].message.content, None
162
  except Exception as e:
163
  return None, str(e)
164
 
 
4
  import tempfile
5
  import os
6
  from typing import Dict
7
+ import openai
8
  from tqdm.auto import tqdm
9
  import plotly.express as px
10
  import plotly.graph_objects as go
 
135
  return responses
136
 
137
  def llamar_api_gpt(descripcion, encuesta, api_key):
138
+ openai.api_key = api_key
139
 
140
  system_content = f"Eres una persona que responde a estas características: \n\n{descripcion}\n\nTu misión consiste en responder a una encuesta. Tu respuesta debe tener este formato. Número de pregunta y opción. Por ejemplo, P1. a), P2. b), etcétera.\n\nEs muy importante que respondas poniéndote en el rol que se te ha dado. No añadas espacios ni saltos de página. La encuesta es: "
141
 
142
  try:
143
+ response = openai.ChatCompletion.create(
144
+ model="gpt-4o",
145
  messages=[
146
+ {"role": "system", "content": system_content},
147
+ {"role": "user", "content": encuesta}
 
 
 
 
 
 
148
  ],
149
  temperature=0,
150
  max_tokens=200,
 
152
  frequency_penalty=0,
153
  presence_penalty=0
154
  )
155
+ return response.choices[0].message['content'], None
156
  except Exception as e:
157
  return None, str(e)
158