Update app.py

app.py

@@ -40,14 +40,23 @@ parser = StrOutputParser()
 # 4. Create chain
 chain = prompt_template | llm | parser
 
-def translate(text, language):
+def translate(text, target_language):
+    source_language = "English"  # Assuming the source language is English
     input_data = {
-        "text": text,
-        "language": language
+        "source_language": source_language,
+        "target_language": target_language,
+        "text": text
     }
     # Use the .invoke() method to properly run the chain
     result = chain.invoke(input_data)
-    return result
+    
+    # Extract the translation from the result
+    if "Assistant: " in result:
+        translation = result.split("Assistant: ")[-1].strip()
+    else:
+        translation = result.strip()
+    
+    return translation
 
 # Gradio interface
 iface = gr.Interface(