Update app.py

app.py

@@ -1,10 +1,12 @@
 import gradio as gr
 import os
-os.environ["KERAS_BACKEND"] = "tensorflow"                                                                           
+os.environ["KERAS_BACKEND"] = "tensorflow"
 import keras
 import keras_nlp
 import utils
 
+from googletrans import Translator
+
 css = """
 html, body {
     margin: 0;
@@ -12,7 +14,6 @@ html, body {
     height: 100%;
     overflow: hidden;
 }
-
 body::before {
     content: '';
     position: fixed;
@@ -23,54 +24,53 @@ body::before {
     background-image: url('https://stsci-opo.org/STScI-01J5E849R5W27ZZ2C3QAE9ET75.png');
     background-size: cover;
     background-repeat: no-repeat;
-    opacity: 0.35;             /* Faint background image */
+    opacity: 0.35;
     background-position: center;
-    z-index: -1;    /* Keep the background behind text */
+    z-index: -1;
 }
 .gradio-container {
     display: flex;
     justify-content: center;
     align-items: center;
-    height: 100vh;  /* Ensure the content is vertically centered */
+    height: 100vh;
 }
 """
 
-#keras.config.set_floatx("bfloat16") uncomment to run at half precision.
+# Модельді жүктеу
 gemma_lm = keras_nlp.models.CausalLM.from_preset("hf://sultan-hassan/CosmoGemma_2b_en")
-#chat = utils.ChatState(gemma_lm) # adding a chat helper to manage the conversation state (modified version from https://github.com/google-gemini/gemma-cookbook/blob/main/Gemma/Keras_Gemma_2_Quickstart_Chat.ipynb)
+chat = utils.ChatState(gemma_lm)
 
-examples = [
-    'What are some common methods for model selection in astrophysics?', 
-    'What could be the cause of remaining excess power in a signal after foreground mitigation?',
-    'What type of coupling in inflationary models can affect the prediction of inflationary parameters?',
-    'Please list three ideas for interesting projects to work on to advance cosmology?',
-    'Please answer the following multiple-choice question.\n\nQuestion: What is the primary goal of calibrating subgrid feedback models in cosmological simulations?\nA) To ensure that simulations accurately reproduce the observed properties of the interstellar medium.\nB) To create a diverse range of galaxy morphologies in the simulations.\nC) To achieve convergence in simulation results across different resolutions and box sizes.\nD) To steer simulations towards producing a broadly realistic galaxy population that is consistent with key observational constraints.\n\n Give only one answer, either A, B, C, or D, but not more than one, and always give an answer.',
-]
+# Аударма объектісі
+translator = Translator()
 
 def launch(message):
+    # 1. Қолданушы сұрағын ағылшыншаға аударамыз (егер сұрақ қазақша болса ғана)
+    detected = translator.detect(message)
+    if detected.lang == 'kk' or detected.lang == 'ky':  # қазақша немесе қырғызша болса
+        message_en = translator.translate(message, src='kk', dest='en').text
+    else:
+        message_en = message
+
+    # 2. Модельге ағылшынша сұрақты жібереміз, чат тарихын қосып
+    response_en = chat.send_message(message_en)
 
-    # Uncomment for QA system without chat history/memory
-    #
-    template = "Instruction:\n{instruction}\n\nResponse:\n{response}"
-    prompt = template.format(
-        instruction=message,                                                                   
-        response="",
-    )
-    out = gemma_lm.generate(prompt, max_length=1024)
-    ind = out.index('\n\nResponse:\n') + len('\n\nResponse:\n')
-    return out[ind:]
-    
-    # This is to convert QA system to a chatbot by adding history to prompt
-    #chat_history = chat.send_message(message) + "\n\n##### History #####\n\n" +  chat.get_history()  
-    #return chat_history
+    # 3. Модельдің ағылшынша жауабын қазақшаға аударамыз
+    response_kk = translator.translate(response_en, src='en', dest='kk').text
+
+    return response_kk
+
+examples = [
+    'Космологиядағы негізгі зерттеу әдістері қандай?',
+    'Инфляциялық модельдерде қандай байланыстар инфляция параметрлерін болжауға әсер етеді?',
+    'Астрономиядағы сигналдағы артық қуаттың себебі неде болуы мүмкін?',
+]
 
 iface = gr.Interface(launch,
                      inputs="text",
                      outputs="text",
                      examples=examples,
                      css=css,
-                     title="Hey I am CosmoGemma 👋 I can answer cosmology questions from astroph.CO research articles. Try me :)",
-                     description="Gemma_2b_en fine-tuned on QA pairs (~3.5k) generated from Cosmology and Nongalactic Astrophysics articles (arXiv astro-ph.CO) from 2018-2022 and tested on QA pairs (~1k) generated from 2023 articles, scoring over 75% accuracy.")
+                     title="CosmoGemma қазақша сөйлейді 👋",
+                     description="CosmoGemma моделін қазақша сұрақтармен пайдаланыңыз, жауаптарды қазақша алыңыз.")
 
 iface.launch()
-