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}] } def chat(text): response = palm.generate_text( **defaults, prompt=f"""Please correct these sentences and rectify any grammar errors. Additionally, when making your corrections, kindly refrain from including quotation marks in your revised sentences. The objective is to enhance the overall clarity and coherence of the paragraph. Sentences: {text}""" ) return response.result app = gr.Interface(fn=chat, inputs=gr.Textbox(lines=10, label="Text", max_lines=10), outputs=gr.Textbox(lines=10, label="Output", max_lines=10, show_copy_button=True), allow_flagging="never") if __name__ == "__main__": app.launch(debug=True)