Update main.py

main.py

@@ -1,9 +1,11 @@
 import json, os
+from io import BytesIO
 import io
 import gradio as gr
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi import FastAPI, Request,  Header, BackgroundTasks, HTTPException, status
 import google.generativeai as genai
+from google.generativeai import types
 import base64
 from collections import defaultdict
 from linebot import LineBotApi, WebhookHandler