mateoluksenberg commited on
Commit
8370975
·
verified ·
1 Parent(s): 8e1d9fb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -10
app.py CHANGED
@@ -206,10 +206,6 @@ EXAMPLES = [
206
  [{"text": "Quiero armar un JSON, solo el JSON sin texto, que contenga los datos de la primera mitad de la tabla de la imagen (las primeras 10 jurisdicciones 901-910). Ten en cuenta que los valores numéricos son decimales de cuatro dígitos. La tabla contiene las siguientes columnas: Codigo, Nombre, Fecha Inicio, Fecha Cese, Coeficiente Ingresos, Coeficiente Gastos y Coeficiente Unificado. La tabla puede contener valores vacíos, en ese caso dejarlos como null. Cada fila de la tabla representa una jurisdicción con sus respectivos valores.", }]
207
  ]
208
 
209
- # Definir la estructura del mensaje utilizando Pydantic
210
- class Message(BaseModel):
211
- text: str
212
- #file: Optional[UploadFile] = None
213
 
214
  # Definir la función simple_chat
215
  def simple_chat(message: Message, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
@@ -269,17 +265,14 @@ def simple_chat(message: Message, temperature: float = 0.8, max_length: int = 40
269
 
270
  return StreamingResponse(generate(), media_type="text/plain")
271
 
272
- # Definir la ruta en FastAPI
273
- @app.post("/chat")
274
- async def chat(message: Message):
275
- return simple_chat(message)
276
-
277
  @app.post("/chat/")
278
  async def test_endpoint(message: dict):
279
  if "text" not in message:
280
  raise HTTPException(status_code=400, detail="Missing 'text' in request body")
 
 
 
281
 
282
- response = {"message": f"Received your message: {message['text']}"}
283
  return simple_chat(message)
284
 
285
  with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo:
 
206
  [{"text": "Quiero armar un JSON, solo el JSON sin texto, que contenga los datos de la primera mitad de la tabla de la imagen (las primeras 10 jurisdicciones 901-910). Ten en cuenta que los valores numéricos son decimales de cuatro dígitos. La tabla contiene las siguientes columnas: Codigo, Nombre, Fecha Inicio, Fecha Cese, Coeficiente Ingresos, Coeficiente Gastos y Coeficiente Unificado. La tabla puede contener valores vacíos, en ese caso dejarlos como null. Cada fila de la tabla representa una jurisdicción con sus respectivos valores.", }]
207
  ]
208
 
 
 
 
 
209
 
210
  # Definir la función simple_chat
211
  def simple_chat(message: Message, temperature: float = 0.8, max_length: int = 4096, top_p: float = 1, top_k: int = 10, penalty: float = 1.0):
 
265
 
266
  return StreamingResponse(generate(), media_type="text/plain")
267
 
 
 
 
 
 
268
  @app.post("/chat/")
269
  async def test_endpoint(message: dict):
270
  if "text" not in message:
271
  raise HTTPException(status_code=400, detail="Missing 'text' in request body")
272
+
273
+ if "file" not in message:
274
+ print("Sin File")
275
 
 
276
  return simple_chat(message)
277
 
278
  with gr.Blocks(css=CSS, theme="soft", fill_height=True) as demo: