eberhenriquez94 commited on
Commit
6bbf7b4
·
verified ·
1 Parent(s): df7a3d1
Files changed (1) hide show
  1. app.py +9 -15
app.py CHANGED
@@ -108,28 +108,22 @@ generation_config = types.GenerateContentConfig(
108
  model_flash_exp = "gemini-2.0-pro-exp-02-05"
109
  model_gemini_ex = "gemini-2.0-flash-thinking-exp-01-21"
110
 
111
- # Función asíncrona para generar contenido con streaming
112
- async def generate_content_streamed(model_name, borrador):
113
- response_stream = await client.aio.models.generate_content_stream(
114
  model=model_name,
115
- contents=borrador, # Se envía solo el borrador; la instrucción está en generation_config
116
  config=generation_config,
117
  )
118
- full_response = ""
119
- async for chunk in response_stream:
120
- full_response += chunk.text
121
- return full_response
122
 
123
- # Combina las respuestas de dos modelos
124
  async def combine_responses(borrador):
125
- flash_task = asyncio.create_task(generate_content_streamed(model_flash_exp, borrador))
126
- gemini_ex_task = asyncio.create_task(generate_content_streamed(model_gemini_ex, borrador))
127
  flash_result = await flash_task
128
  gemini_ex_result = await gemini_ex_task
129
- combined_result = (
130
- f"**Google Gemini flash-exp**:\n{flash_result}\n\n"
131
- f"**Google gemini-exp-1206**:\n{gemini_ex_result}"
132
- )
133
  return combined_result
134
 
135
  # Función asíncrona principal de predicción
 
108
  model_flash_exp = "gemini-2.0-pro-exp-02-05"
109
  model_gemini_ex = "gemini-2.0-flash-thinking-exp-01-21"
110
 
111
+ # Función ASÍNCRONA para generar contenido (SIN STREAMING)
112
+ async def generate_content_async(model_name, borrador):
113
+ response = await client.aio.models.generate_content( # Usamos la versión async de generate_content
114
  model=model_name,
115
+ contents=borrador,
116
  config=generation_config,
117
  )
118
+ return response.text
 
 
 
119
 
120
+ # Combina las respuestas de dos modelos (ASÍNCRONO)
121
  async def combine_responses(borrador):
122
+ flash_task = asyncio.create_task(generate_content_async(model_flash_exp, borrador)) # Llama a la versión async SIN streaming
123
+ gemini_ex_task = asyncio.create_task(generate_content_async(model_gemini_ex, borrador)) # Llama a la versión async SIN streaming
124
  flash_result = await flash_task
125
  gemini_ex_result = await gemini_ex_task
126
+ combined_result = flash_result + "\n\n" + gemini_ex_result
 
 
 
127
  return combined_result
128
 
129
  # Función asíncrona principal de predicción