bugfix

app.py

@@ -18,7 +18,7 @@ def pdf_to_text_with_layout(pdf_file):
 
 def clean_text(text):
     # Remove non-ASCII and control characters
-    return ''.join(c for c in text if c.isprintable() and ord(c) < 65536)
+    return ''.join(c for c in text if c.is printable() and ord(c) < 65536)
 
 def text_to_word_with_formatting(text, word_path):
     doc = Document()

langchain_pipeline.py

@@ -10,18 +10,20 @@ import vertexai
 from vertexai.generative_models import GenerativeModel, Part, FinishReason
 import vertexai.preview.generative_models as generative_models
 
-def generate():
+def generate(document_parts, prompt_text):
     vertexai.init(project="akroda", location="us-central1")
     model = GenerativeModel("gemini-1.5-pro-001")
     responses = model.generate_content(
-        [document1, document2, document3, document4, document5, document6, document7, document8, document9, text1],
+        document_parts,
         generation_config=generation_config,
         safety_settings=safety_settings,
         stream=True,
     )
 
+    response_text = ""
     for response in responses:
-        print(response.text, end="")
+        response_text += response.text
+    return response_text
 
 document1 = Part.from_data(
     mime_type="application/pdf",
@@ -61,9 +63,9 @@ What is the minimum amount overdrawn to incur a fee?: ${min_overdrawn_fee}
 What is the minimum transaction amount to trigger an overdraft?: ${min_transaction_overdraft}
 
 Please output in the following format:
-{entire updated disclosure text with changes bolded}
+{{entire updated disclosure text with changes bolded}}
 ------
-{reasons for each change listed and cases cited}
+{{reasons for each change listed and cases cited}}
 """
 
 prompt = PromptTemplate(
@@ -71,9 +73,21 @@ prompt = PromptTemplate(
     template=text1,
 )
 
+# Placeholder values for the variables used in prompt formatting
+legal_cases_context = "Provide the legal context here..."
+disclosure_text = "Include the initial disclosure text here..."
+balance_type = "available balance"
+apsn_transactions = "yes"
+max_fees_per_day = 3
+min_overdrawn_fee = 5
+min_transaction_overdraft = 1
+
+# Base64 encode the disclosure text
+encoded_disclosure_text = base64.b64encode(disclosure_text.encode()).decode()
+
 val = prompt.format(
     context=legal_cases_context,
-    disclosure=disclosure_text,
+    disclosure=encoded_disclosure_text,
     balance_type=balance_type,
     apsn_transactions=apsn_transactions,
     max_fees_per_day=max_fees_per_day,
@@ -95,5 +109,6 @@ safety_settings = {
 }
 
 def pipeline(file, model_name, balance_type, apsn_transactions, max_fees_per_day, min_overdrawn_fee, min_transaction_overdraft):
-    generate()
-    return chat_response.content
+    document_parts = [document1, document2, document3, document4, document5, document6, document7, document8, document9]
+    response_text = generate(document_parts, val)
+    return response_text