import gradio as gr
import google.generativeai as palm
import os

# Get your own API key on https://developers.generativeai.google/products/palm
api_key = os.environ["api_key"]
palm.configure(api_key=api_key)

defaults = {
    'model': 'models/text-bison-001',
    'temperature': 0.7,
    'candidate_count': 1,
    'top_k': 40,
    'top_p': 0.95,
    'max_output_tokens': 1024,
    'stop_sequences': [],
    'safety_settings': [
        {"category": "HARM_CATEGORY_DEROGATORY", "threshold": 3},
        {"category": "HARM_CATEGORY_TOXICITY", "threshold": 3},
        {"category": "HARM_CATEGORY_VIOLENCE", "threshold": 3},
        {"category": "HARM_CATEGORY_SEXUAL", "threshold": 3},
        {"category": "HARM_CATEGORY_MEDICAL", "threshold": 3},
        {"category": "HARM_CATEGORY_DANGEROUS", "threshold": 3}]

}

with open("example.txt") as f:
    example = f.read()

with gr.Blocks() as app:
    def chat(text):
        try:
            response = palm.generate_text(
            **defaults,
            prompt = f"""Rewrite the following sentence to fix the grammar issues and correct the sentence.
                    {example}
                    input: {text}
                    fixed"""
            )
            return response.result
        
        except:
            return "I am not able to correct the sentence given. Please try again."
        
    with gr.Column():
        text = gr.Textbox(lines=4, label="Text", max_lines=4, placeholder="Write something awesome. It will be corrected automatically.")

    with gr.Column():
        output = gr.Textbox(lines=4, label="Output", max_lines=4, show_copy_button=True)
        
        with gr.Row():
            clr_btn = gr.ClearButton([text, output])
            btn = gr.Button("Submit", variant="primary")
            btn.click(fn=chat, inputs=text, outputs=output)

    gr.Examples(
        ["Do you ready to lose?", "He don't like vegetables", "Can I gets a glass of water?"],
        text
    )

    
if __name__ == "__main__":
    app.launch()