Antigravity Bot commited on
Commit
0865bea
1 Parent(s): cb415e1

Add debug prints and sanitize messages

Browse files
Files changed (1) hide show
  1. app.py +17 -1
app.py CHANGED
@@ -12,8 +12,24 @@ def predict(message, history):
12
 
13
  # Criamos a lista de mensagens adicionando a mensagem atual ao hist贸rico
14
  # O history vem do Gradio, o message 茅 o input atual
15
- messages = history + [{"role": "user", "content": message}]
 
16
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
17
  # Gera a resposta
18
  results = pipe(messages, max_new_tokens=512)
19
 
 
12
 
13
  # Criamos a lista de mensagens adicionando a mensagem atual ao hist贸rico
14
  # O history vem do Gradio, o message 茅 o input atual
15
+ if message is None:
16
+ message = ""
17
 
18
+ print(f"DEBUG INPUT: message={repr(message)}")
19
+ print(f"DEBUG INPUT: history={repr(history)}")
20
+
21
+ new_messages = history + [{"role": "user", "content": message}]
22
+
23
+ # Sanitize messages to ensure no None content
24
+ messages = []
25
+ for msg in new_messages:
26
+ content = msg.get("content")
27
+ if content is None:
28
+ content = ""
29
+ messages.append({"role": msg.get("role"), "content": content})
30
+
31
+ print(f"DEBUG FINAL MESSAGES: {messages}")
32
+
33
  # Gera a resposta
34
  results = pipe(messages, max_new_tokens=512)
35