Update app.py

app.py

@@ -8,6 +8,8 @@ anyscale_base_url = "https://api.endpoints.anyscale.com/v1"
 multilingual_embeddings = embedding_functions.SentenceTransformerEmbeddingFunction(model_name="jost/multilingual-e5-base-politics-de")
 
 def predict(api_key, user_input, model1, model2, prompt_manipulation=None, direct_steering_option=None):
+    print(prompt_manipulation)
+    print(direct_steering_option)
     # client = chromadb.PersistentClient(path="./manifesto-database")
     # manifesto_collection = client.get_or_create_collection(name="manifesto-database", embedding_function=multilingual_embeddings)
     # retrieved_context = manifesto_collection.query(query_texts=[user_input], n_results=3, where={"ideology": "Authoritarian-right"})
@@ -86,7 +88,7 @@ def main():
             output1 = gr.Textbox(label="Model 1 Response")
             output2 = gr.Textbox(label="Model 2 Response")
         
-        submit_btn.click(fn=predict, inputs=[api_key_input, user_input, model_selector1, model_selector2], outputs=[output1, output2])
+        submit_btn.click(fn=predict, inputs=[api_key_input, user_input, model_selector1, model_selector2, prompt_manipulation, direct_steering_option], outputs=[output1, output2])
 
     demo.launch()