Sayandip/Bengali_Text_Simplification

1import gradio as gr2from google import genai3from google.genai import types4import os5 6# Get the Gemini API key from environment variable7api_key = os.getenv("GEMINI_API_KEY")8if not api_key:9    raise ValueError("GEMINI_API_KEY environment variable not set.")10 11# Create a Gemini client using the API key12client = genai.Client(api_key=api_key)13 14def simplify_bengali_text(15    message,16    history: list[tuple[str, str]],17    system_message,18    max_tokens,19    temperature,20    top_p,21):22    full_prompt = (23        f"এই লেখাটি সহজ বাংলায় লিখো যাতে সাধারণ পাঠক সহজেই বুঝতে পারে। "24        f"এটি কোনো সারাংশ, তালিকা, বা বিন্দুযুক্ত আকারে লেখা হোক না। "25        f"মূল বক্তব্য যেন অপরিবর্তিত থাকে, কিন্তু ভাষাটি সহজ ও প্রাঞ্জল হয়।\n\n"26        f"মূল পাঠঃ\n{message}"27    )28 29    part = types.Part.from_text(text=full_prompt)30 31    contents = [32        types.Content(33            role="user",34            parts=[part]35        )36    ]37 38    config = types.GenerateContentConfig(39        temperature=temperature,40        top_p=top_p,41        max_output_tokens=max_tokens,42        response_modalities=["TEXT"],43        safety_settings=[44            types.SafetySetting(category="HARM_CATEGORY_HATE_SPEECH", threshold="OFF"),45            types.SafetySetting(category="HARM_CATEGORY_DANGEROUS_CONTENT", threshold="OFF"),46            types.SafetySetting(category="HARM_CATEGORY_SEXUALLY_EXPLICIT", threshold="OFF"),47            types.SafetySetting(category="HARM_CATEGORY_HARASSMENT", threshold="OFF"),48        ]49    )50 51    response = ""52    for chunk in client.models.generate_content_stream(53        model="gemini-2.0-flash-001",54        contents=contents,55        config=config,56    ):57        if hasattr(chunk, "text"):58            response += chunk.text59            yield response60 61# Gradio UI62demo = gr.ChatInterface(63    simplify_bengali_text,64    additional_inputs=[65        gr.Textbox(value="Please simplify the given Bengali text in same meaning but in easy undestandable Bengali that can understood by masses in recent times.", label="System message"),66        gr.Slider(minimum=1, maximum=8192, value=2048, step=1, label="Max new tokens"),67        gr.Slider(minimum=0.1, maximum=2.0, value=1.0, step=0.1, label="Temperature"),68        gr.Slider(minimum=0.1, maximum=1.0, value=0.95, step=0.05, label="Top-p"),69    ],70)71 72if __name__ == "__main__":73    demo.launch()