Update app.py

app.py

@@ -1,4 +1,12 @@
 import gradio as gr
+import google.generativeai as genai
+
+import os 
+token=os.environ.get("TOKEN") 
+genai.configure(api_key=token)
+
+
+model_simple = genai.GenerativeModel(model_name="gemini-pro")
 
 def generate_data(subject, input_field, difficulty):
     # Traiter les données ici, par exemple imprimer les données
@@ -12,8 +20,10 @@ def generate_data(subject, input_field, difficulty):
     
     """
 
-    
-    return q
+    response = model_simple.generate_content(q)
+    e = response.text
+    print(e)
+    return e
 
 # Créer l'interface utilisateur Gradio