Update main.py

main.py

@@ -30,6 +30,21 @@ load_dotenv()
 app = Flask(__name__)
 cors = CORS(app)
 
+class StreamLitResponse(ResponseParser):
+        def __init__(self,context) -> None:
+              super().__init__(context)
+        def format_dataframe(self,result):
+               st.dataframe(result['value'])
+               return
+        def format_plot(self,result):
+               st.image(result['value'])
+               return
+        def format_other(self, result):
+               st.write(result['value'])
+               return
+
+gemini_api_key = os.environ['Gemini']
+
 @app.route("/", methods=["GET"])
 def home():