test genai

main.py

@@ -6,15 +6,15 @@ import os
 
 import mesop as me
 import mesop.labs as mel
+import google.generativeai as genai
 
-API_KEY = os.getenv("API_KEY")
-
+MY_API_KEY = os.getenv("API_KEY")
+genai.configure(api_key=MY_API_KEY)
+model = genai.GenerativeModel('models/gemini-2.0-flash')
 
 def on_load(e: me.LoadEvent):
   me.set_theme_mode("system")
 
-
-
 @me.page(
   security_policy=me.SecurityPolicy(
     allowed_iframe_parents=["https://huggingface.co"]
@@ -27,11 +27,11 @@ def page():
 
 
 def transform(input: str, history: list[mel.ChatMessage]):
-  for line in random.sample(LINES, random.randint(3, len(LINES) - 1)):
-    time.sleep(0.3)
-    yield line + " "
+  model.generate_content(input)
+  # for line in random.sample(LINES, random.randint(3, len(LINES) - 1)):
+  #   time.sleep(0.3)
+  #   yield line + " "
 
-print(API_KEY)
 
 LINES = [
   "Mesop is a Python-based UI framework designed to simplify web UI development for engineers without frontend experience.",

requirements.txt

@@ -1,4 +1,5 @@
 mesop
 gunicorn
 pandas
-numpy
+numpy
+google-generativeai