Update app.py

app.py

@@ -57,7 +57,7 @@ def predict(
         manifesto_collection = client.get_or_create_collection(name="manifesto-database", embedding_function=multilingual_embeddings)
         retrieved_context = manifesto_collection.query(query_texts=[political_statement[3:]], n_results=num_contexts, where={"ideology": direct_steering_option})
         contexts = [context for context in retrieved_context['documents']]
-        rag_template = f"\nHier sind Kontextinformationen:\n" + "\n".join([f"{context}" for context in contexts])
+        rag_template = f"\nHier sind Kontextinformationen:\n" + "\n".join([f"{context}" for context in contexts[0]])
         
         prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement[3:], rag_template=rag_template)
         print(prompt)
@@ -83,7 +83,7 @@ def predict(
         impersonation_template = ""
         answer_option_template = f"{test_format[ideology_test]}"
         rag_template = ""
-        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement, rag_template=rag_template)
+        prompt = prompt_template.format(impersonation_template=impersonation_template, answer_option_template=answer_option_template, statement=political_statement[3:], rag_template=rag_template)
         print(prompt)
 
     client = OpenAI(base_url=togetherai_base_url, api_key=togetherai_api_key)