Updates

app.py

@@ -105,6 +105,7 @@ class BasicAgent:
             file_extension = state["file_path"].split(".")[1]
         else:
             file_extension = None
+            file_name = None
         
         prompt = f"""
         You are a helpful assistant. 
@@ -150,7 +151,7 @@ class BasicAgent:
             msg = "File downloaded successfully!!"
             print(msg)
             file = response.content
-            b64_file = base64.b64encode(state["file"]).decode("utf-8")
+            b64_file = base64.b64encode(file).decode("utf-8")
         else:
             msg = "There was an error downloading the file."
             print(msg)
@@ -197,6 +198,8 @@ class BasicAgent:
             message = {"type": "media", "data": encoded_file,  # Use base64 string directly
             "mime_type": "audio/mpeg",
         }
+        else:
+            message = {"type": "text", "text": "The file is not supported."}
 
         message_local = HumanMessage(
             content=[
@@ -209,7 +212,7 @@ class BasicAgent:
             model="gemini-2.0-flash",
             temperature=0)
 
-        response = llm.invoke(message_local)
+        response = llm.invoke([message_local])
 
         return response