Update app.py

app.py

@@ -10,6 +10,7 @@ import json
 import base64 # <-- ADDED IMPORT for image handling
 import re
 import time
+import pandas as pd
 # Attempt to login using environment token
 try:
     HF_TOKEN = os.environ.get("HUGGINGFACE_TOKEN")
@@ -287,10 +288,29 @@ def my_agent_logic(task_id: str, question: str, files_metadata: list = None): #
             except Exception as e_txt:
                 print(f"Error decoding text file {file_name} for task {task_id}: {e_txt}")
                 gemini_parts.append({"text": user_question_text + f"\n[Agent note: A text file '{file_name}' was associated but could not be decoded: {e_txt}]"})
-        else: # Other file types, just mention them
-            user_question_text += f"\n\nNote: A file named '{file_name}' (type: {detected_mime_type or 'unknown'}) is associated with this question. Its content is not directly viewable in this text prompt."
-            gemini_parts.append({"text": user_question_text})
-            print(f"Noted non-image/text file {file_name} ({detected_mime_type}) in Gemini prompt for task {task_id}.")
+        ###########################################################################################################################
+        elif detected_mime_type in [
+            "application/vnd.openxmlformats-officedocument.spreadsheetml.sheet", 
+            "application/vnd.ms-excel", 
+            "text/csv"]:
+            try:
+                from io import BytesIO
+                if "csv" in detected_mime_type:
+                    df = pd.read_csv(BytesIO(file_content_bytes))
+                else:
+                    df = pd.read_excel(BytesIO(file_content_bytes))
+        
+                preview = df.head(10).to_string(index=False)
+                user_question_text += f"\n\nContent preview from the attached spreadsheet '{file_name}':\n{preview}"
+                gemini_parts.append({"text": user_question_text})
+                print(f"Added spreadsheet content preview for {file_name} to Gemini prompt.")
+            except Exception as e_xls:
+                print(f"Error reading spreadsheet file {file_name} for task {task_id}: {e_xls}")
+                user_question_text += f"\n\n[Agent note: Unable to parse spreadsheet '{file_name}': {e_xls}]"
+                gemini_parts.append({"text": user_question_text})
+
+
+        ###########################################################################################################################
     else: # No file content fetched or no files associated
         gemini_parts.append({"text": user_question_text})