BATUTO-ART commited on
Commit
52f9716
verified
1 Parent(s): 66f02ff

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -42,10 +42,10 @@ VOYEUR_SPECIALIST_CONFIG = {
42
  ],
43
  }
44
 
45
- # Inicializaci贸n del cliente SambaNova (configura tus credenciales y URL)
46
  client = SambaNova(
47
- api_key=os.getenv("SAMBANOVA_API_KEY"), # Aseg煤rate de exportar esta variable de entorno
48
- base_url=os.getenv("SAMBANOVA_BASE_URL") # Ejemplo: "https://api.sambanova.ai/v1"
49
  )
50
 
51
  # ============================================
@@ -91,12 +91,13 @@ def build_prompt(user_text: str) -> str:
91
  return SYSTEM_PROMPT_BASE + "\n\nUSER:\n" + user_text
92
 
93
  def call_model(prompt: str) -> str:
94
- # Llamada al modelo SambaNova para obtener la respuesta real
95
  response = client.chat.completions.create(
96
  model=MODEL_NAME,
97
- messages=[{"role": "system", "content": SYSTEM_PROMPT_BASE}, {"role": "user", "content": prompt}],
 
 
 
98
  )
99
- # Extraer el texto de la respuesta
100
  return response.choices[0].message["content"]
101
 
102
  def infer(user_text: str) -> str:
@@ -197,12 +198,11 @@ with gr.Blocks(title="Voyeur Prompt Generator") as demo:
197
  out = gr.Textbox(
198
  label="Prompt generado (Ingl茅s)",
199
  lines=8,
200
- show_copy_button=True
201
  )
202
 
203
  btn.click(fn=infer, inputs=inp, outputs=out)
204
 
205
-
206
  if __name__ == "__main__":
207
  demo.launch()
208
 
 
42
  ],
43
  }
44
 
45
+ # Cliente SambaNova inicializado con variables de entorno
46
  client = SambaNova(
47
+ api_key=os.getenv("SAMBANOVA_API_KEY"),
48
+ base_url=os.getenv("SAMBANOVA_BASE_URL")
49
  )
50
 
51
  # ============================================
 
91
  return SYSTEM_PROMPT_BASE + "\n\nUSER:\n" + user_text
92
 
93
  def call_model(prompt: str) -> str:
 
94
  response = client.chat.completions.create(
95
  model=MODEL_NAME,
96
+ messages=[
97
+ {"role": "system", "content": SYSTEM_PROMPT_BASE},
98
+ {"role": "user", "content": prompt}
99
+ ],
100
  )
 
101
  return response.choices[0].message["content"]
102
 
103
  def infer(user_text: str) -> str:
 
198
  out = gr.Textbox(
199
  label="Prompt generado (Ingl茅s)",
200
  lines=8,
201
+ buttons=["copy"]
202
  )
203
 
204
  btn.click(fn=infer, inputs=inp, outputs=out)
205
 
 
206
  if __name__ == "__main__":
207
  demo.launch()
208