ORromu commited on
Commit
ff94a5a
·
verified ·
1 Parent(s): 2ece109

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +14 -0
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})