Commit eb5a8a6

@@ -130,10 +130,10 @@ async def single_gemini_context(client: Client, message: Message, app: genai.Cli
     if info["mtype"] not in ["text", "photo", "audio", "voice", "video", "document"]:
         return {}
     # gemini has built-in support for these extensions
-    gemini_extensions = [".pdf", ".js", ".py", ".txt", ".html", ".css", ".md", ".csv", ".xml", ".rtf", ".mp3", ".wav", ".ogg", ".aac", ".flac", ".jpg", ".jpeg", ".webp", ".png", ".heic", ".heif"]
+    gemini_extensions = [".pdf", ".html", ".css", ".csv", ".xml", ".rtf", ".mp3", ".wav", ".ogg", ".aac", ".flac", ".jpg", ".jpeg", ".webp", ".png", ".heic", ".heif"]
     # gemini has built-in support for these mime types
     gemini_mime_types = ["application/pdf", "application/x-javascript", "audio/ogg", "audio/mp4", "image/jpeg", "image/png", "image/webp", "image/heic", "image/heif"]
-    extra_txt_extensions = [".sh", ".json"]  # also treat these as txt file
+    txt_extensions = [".txt", ".js", ".py", ".md", ".sh", ".json"]  # treat these as txt file
     extra_markdown_extensions = [".docx", ".pptx", ".xls", ".xlsx"]  # convert to markdown
 
     messages = await client.get_media_group(message.chat.id, message.id) if message.media_group_id else [message]
@@ -152,7 +152,7 @@ async def single_gemini_context(client: Client, message: Message, app: genai.Cli
                     parts.append(Part.from_uri(file_uri=upload.uri, mime_type=upload.mime_type))
                 Path(fpath).unlink(missing_ok=True)
             elif info["mtype"] == "document":
-                if info["mime_type"].startswith("text/") or Path(info["file_name"]).suffix in extra_txt_extensions:
+                if info["mime_type"].startswith("text/") or Path(info["file_name"]).suffix in txt_extensions:
                     res = await base64_media(client, msg)
                     parts.append(Part.from_text(text=f"[filename]: {info['file_name']}\n[file content]:\n{res['value'].strip()}"))
                 if Path(info["file_name"]).suffix in extra_markdown_extensions: