Changes

app.py

@@ -17,7 +17,8 @@ def respond(
     temperature,
     top_p,
     sourceLanguage,
-    targetLanguage
+    targetLanguage,
+    useGoogle
 ):
     toFinnish=targetLanguage=='Finnish' and sourceLanguage=="English"
     fromFinnish=targetLanguage=='English' and sourceLanguage=="Finnish"
@@ -27,11 +28,36 @@ def respond(
     #candidateClient=finnishToEnglishClient if(fromFinnish)else googleClient 
 
     print("the text",message, 'the source',sourceLanguage,"the target",targetLanguage)
- 
-    response=''
-
+  
+    if(useGoogle):
+        text="Translate to "f"{targetLanguage}: "f"{message}"
+        googleResponse=  googleClient.text_generation(
+            text
+        )
 
+        return googleResponse
+    
     if(toFinnish):
+        fiResponse=  englishToFinnishClient.translation(
+            message,
+            #max_new_tokens=100
+            tgt_lang= targetLanguage,
+            src_lang=sourceLanguage  
+        )
+
+        return fiResponse.translation_text
+    
+    if(fromFinnish):
+        fiResponse=finnishToEnglishClient.translation(
+            message,
+            tgt_lang= targetLanguage,
+            src_lang=sourceLanguage  
+        )
+
+        return fiResponse.translation_text
+
+
+    """ if(toFinnish):
         fiResponse=  englishToFinnishClient.translation(
         message,
         #max_new_tokens=100
@@ -68,7 +94,7 @@ def respond(
 
     print("the response",response)
 
-    response
+    response """
         
 
 """
@@ -89,6 +115,7 @@ demo = gr.ChatInterface(
         ),
         gr.Textbox(value="Finnish", label="Source Language"),
         gr.Textbox(value="English", label="Target Language"),
+        gr.Textbox(value="false", label="Use Google"),
     ],
 )