import gradio as gr
import google.generativeai as genai
import os 
token=os.environ.get("TOKEN")

description_french = " Esseu je dois encore expliquer ?"

genai.configure(
    api_key=token
)
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





    

iface = gr.Interface(
    fn=gpt_francais,
    title="Mariam-French",
    description=description_french,
    inputs=[gr.Textbox(label="Sujet:", lines=3),gr.Radio(["Etaye","refute"]),gr.Radio(["Victor Hugo","Marcel Proust","Albert Camus","Marguerite Duras","Gustave Flaubert","Prof"],label="Style d'ecriture")],
    outputs=gr.Textbox(),
)
iface.launch()