Update app.py

app.py

@@ -32,7 +32,7 @@ def create_persona(text):
                 model="gpt-3.5-turbo",
                 messages=[
                     {"role": "system", "content": "You are a knowledgeable assistant."},
-                    {"role": "user", "content": text},
+                    {"role": "user", "content": f"Create a persona based on this text: {text}"},
                 ]
             )
             return response['choices'][0]['message']['content']
@@ -64,21 +64,20 @@ def call_openai_api(persona, user_prompt):
 
 def pdf_to_text(file, user_prompt):
     z = zipfile.ZipFile(file.name, 'r')
-    texts = []
+    aggregated_text = ''
     for filename in z.namelist():
         if filename.endswith('.pdf'):
             pdf_file_data = z.read(filename)
             pdf_file_io = io.BytesIO(pdf_file_data)
             pdf = PdfReader(pdf_file_io)
-            text = ''
             for page in pdf.pages:
-                text += page.extract_text()
-            # Create persona from text
-            persona = create_persona(text)
-            # Using OpenAI API
-            response = call_openai_api(persona, user_prompt)
-            texts.append(response)
-    return '\n'.join(texts)
+                aggregated_text += page.extract_text()
+    # Create a single persona from all text
+    persona = create_persona(aggregated_text)
+    # Using OpenAI API
+    response = call_openai_api(persona, user_prompt)
+    return response
+
 
 iface = gr.Interface(
     fn=pdf_to_text,