Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -70,6 +70,20 @@ class BasicAgent:
|
|
| 70 |
print(f"Error processing audio: {e}")
|
| 71 |
return f"Error: {e}"
|
| 72 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 73 |
else:
|
| 74 |
messages = [HumanMessage(content=full_prompt)]
|
| 75 |
messages = self.graph.invoke({"messages": messages})
|
|
|
|
| 70 |
print(f"Error processing audio: {e}")
|
| 71 |
return f"Error: {e}"
|
| 72 |
|
| 73 |
+
elif file_type == "xlsx":
|
| 74 |
+
try:
|
| 75 |
+
encoded_file = base64.b64encode(file_bytes).decode("utf-8")
|
| 76 |
+
xlsx_mime_type = "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet"
|
| 77 |
+
|
| 78 |
+
messages = [HumanMessage(content=[
|
| 79 |
+
{"type": "text", "text": question},
|
| 80 |
+
{
|
| 81 |
+
"type": "media",
|
| 82 |
+
"data": encoded_file,
|
| 83 |
+
"mime_type": xlsx_mime_type,
|
| 84 |
+
}
|
| 85 |
+
])]
|
| 86 |
+
|
| 87 |
else:
|
| 88 |
messages = [HumanMessage(content=full_prompt)]
|
| 89 |
messages = self.graph.invoke({"messages": messages})
|