yodafy3

app.py

@@ -5,12 +5,12 @@ from langchain.chains.llm import LLMChain
 from langchain.chains.constitutional_ai.base import ConstitutionalChain
 from langchain.chains.constitutional_ai.models import ConstitutionalPrinciple
 
-def yodafy(sentence):
+def yodafy(sentence, selection):
 
     llm = OpenAI(temperature=.8)
     prompt = PromptTemplate(
-        input_variables=["sentence"],
-        template="Rewrite the sentence inside <> as Master Yoda. Sentence: <{sentence}>",
+        input_variables=["sentence", "selection"],
+        template=f"{selection} {sentence} as Master Yoda would.",
     )
 
     chain = LLMChain(llm=llm, prompt=prompt)
@@ -34,8 +34,10 @@ def yodafy(sentence):
         verbose=True,
     )
 
-    return constitutional_chain.run(sentence=sentence)
+    return constitutional_chain.run(sentence=sentence, selection=selection)
 
 
-iface = gr.Interface(fn=yodafy, inputs="text", outputs="text")
+iface = gr.Interface(fn=yodafy, 
+                     inputs=["text", gr.inputs.Radio(["Rewrite", "Reply"])], 
+                     outputs="text")
 iface.launch()