Update app.py

app.py

@@ -6,14 +6,62 @@ token=os.environ.get("TOKEN")
 genai.configure(
     api_key=token
 )
-model = genai.GenerativeModel(
-    model_name='gemini-1.5-flash-latest'
-)
+generation_config = {
+  "temperature": 1,
+  "top_p": 0.95,
+  "top_k": 64,
+  "max_output_tokens": 8192,
+}
+
+safety_settings = [
+  {
+    "category": "HARM_CATEGORY_HARASSMENT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_HATE_SPEECH",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
+    "threshold": "BLOCK_NONE"
+  },
+  {
+    "category": "HARM_CATEGORY_DANGEROUS_CONTENT",
+    "threshold": "BLOCK_NONE"
+  },
+]
+
+model = genai.GenerativeModel(model_name="gemini-1.5-flash-latest",
+                              generation_config=generation_config,
+                              safety_settings=safety_settings)
+
+
+def gpt_francais(french_prompt,choix,autheur):
+    
+    if not french_prompt:
+        return "Veuillez saisir un thème ."
+
+    if choix == "discuter":
+
+    	de = """ Je veux faire mon travail de français  de niveau lycé sous la forme d'une discussion  argumentatif. La question du travail est la suivante: "{french_prompt}". tu devras discuter ce thème. jai besoin dune introduction(avec une problématique et l'annonce du plan), de 3 arguments(avec explication et example et illustration) pour la premiere partie. idem pour la deuxieme partie(3 argument pour lantithese). je veux une phrase de transition entre les deux partie  et a la fin 'une conclusion avec une ouverture(donné sous forme de phrase interogative ) je veux que tu utilise le style décriture de :{autheur} """
+    	answer = model.generate_content(french_prompt)
+        #return response.text
+    	print(answer)
+    
+    	return answer.text
+    else:
+    	fi_prompt = f"""Je veux faire mon travail de français  de niveau lycé sous la forme d'un travail argumentatif. La question du travail est la suivante: "{french_prompt}". tu devras {choix} ce thème. jai besoin dune introduction(avec une problématique et l'annonce du plan), de 3 arguments(avec explication et example et illustration)  et d'une conclusion avec une ouverture(donné sous forme de phrase interogative ) je veux que tu utilise le style décriture de :{autheur} """   
+    	#answer = res(fi_prompt)
+    	response = model.generate_content(french_prompt)
+    	print(response)
+    	return response.text
+
+
+
+
 
-def greet(name):
-    print(name)
-    response = model.generate_content(name)
-    return response.text
+    
 
 iface = gr.Interface(fn=greet, inputs=gr.Textbox(label="Question:", lines=4), outputs="text")
 iface.launch()