Adding PDF Functionality

.chainlit/translations/en-US.json

@@ -0,0 +1,229 @@
+{
+    "components": {
+        "atoms": {
+            "buttons": {
+                "userButton": {
+                    "menu": {
+                        "settings": "Settings",
+                        "settingsKey": "S",
+                        "APIKeys": "API Keys",
+                        "logout": "Logout"
+                    }
+                }
+            }
+        },
+        "molecules": {
+            "newChatButton": {
+                "newChat": "New Chat"
+            },
+            "tasklist": {
+                "TaskList": {
+                    "title": "\ud83d\uddd2\ufe0f Task List",
+                    "loading": "Loading...",
+                    "error": "An error occurred"
+                }
+            },
+            "attachments": {
+                "cancelUpload": "Cancel upload",
+                "removeAttachment": "Remove attachment"
+            },
+            "newChatDialog": {
+                "createNewChat": "Create new chat?",
+                "clearChat": "This will clear the current messages and start a new chat.",
+                "cancel": "Cancel",
+                "confirm": "Confirm"
+            },
+            "settingsModal": {
+                "settings": "Settings",
+                "expandMessages": "Expand Messages",
+                "hideChainOfThought": "Hide Chain of Thought",
+                "darkMode": "Dark Mode"
+            },
+            "detailsButton": {
+                "using": "Using",
+                "used": "Used"
+            },
+            "auth": {
+                "authLogin": {
+                    "title": "Login to access the app.",
+                    "form": {
+                        "email": "Email address",
+                        "password": "Password",
+                        "noAccount": "Don't have an account?",
+                        "alreadyHaveAccount": "Already have an account?",
+                        "signup": "Sign Up",
+                        "signin": "Sign In",
+                        "or": "OR",
+                        "continue": "Continue",
+                        "forgotPassword": "Forgot password?",
+                        "passwordMustContain": "Your password must contain:",
+                        "emailRequired": "email is a required field",
+                        "passwordRequired": "password is a required field"
+                    },
+                    "error": {
+                        "default": "Unable to sign in.",
+                        "signin": "Try signing in with a different account.",
+                        "oauthsignin": "Try signing in with a different account.",
+                        "redirect_uri_mismatch": "The redirect URI is not matching the oauth app configuration.",
+                        "oauthcallbackerror": "Try signing in with a different account.",
+                        "oauthcreateaccount": "Try signing in with a different account.",
+                        "emailcreateaccount": "Try signing in with a different account.",
+                        "callback": "Try signing in with a different account.",
+                        "oauthaccountnotlinked": "To confirm your identity, sign in with the same account you used originally.",
+                        "emailsignin": "The e-mail could not be sent.",
+                        "emailverify": "Please verify your email, a new email has been sent.",
+                        "credentialssignin": "Sign in failed. Check the details you provided are correct.",
+                        "sessionrequired": "Please sign in to access this page."
+                    }
+                },
+                "authVerifyEmail": {
+                    "almostThere": "You're almost there! We've sent an email to ",
+                    "verifyEmailLink": "Please click on the link in that email to complete your signup.",
+                    "didNotReceive": "Can't find the email?",
+                    "resendEmail": "Resend email",
+                    "goBack": "Go Back",
+                    "emailSent": "Email sent successfully.",
+                    "verifyEmail": "Verify your email address"
+                },
+                "providerButton": {
+                    "continue": "Continue with {{provider}}",
+                    "signup": "Sign up with {{provider}}"
+                },
+                "authResetPassword": {
+                    "newPasswordRequired": "New password is a required field",
+                    "passwordsMustMatch": "Passwords must match",
+                    "confirmPasswordRequired": "Confirm password is a required field",
+                    "newPassword": "New password",
+                    "confirmPassword": "Confirm password",
+                    "resetPassword": "Reset Password"
+                },
+                "authForgotPassword": {
+                    "email": "Email address",
+                    "emailRequired": "email is a required field",
+                    "emailSent": "Please check the email address {{email}} for instructions to reset your password.",
+                    "enterEmail": "Enter your email address and we will send you instructions to reset your password.",
+                    "resendEmail": "Resend email",
+                    "continue": "Continue",
+                    "goBack": "Go Back"
+                }
+            }
+        },
+        "organisms": {
+            "chat": {
+                "history": {
+                    "index": {
+                        "showHistory": "Show history",
+                        "lastInputs": "Last Inputs",
+                        "noInputs": "Such empty...",
+                        "loading": "Loading..."
+                    }
+                },
+                "inputBox": {
+                    "input": {
+                        "placeholder": "Type your message here..."
+                    },
+                    "speechButton": {
+                        "start": "Start recording",
+                        "stop": "Stop recording"
+                    },
+                    "SubmitButton": {
+                        "sendMessage": "Send message",
+                        "stopTask": "Stop Task"
+                    },
+                    "UploadButton": {
+                        "attachFiles": "Attach files"
+                    },
+                    "waterMark": {
+                        "text": "Built with"
+                    }
+                },
+                "Messages": {
+                    "index": {
+                        "running": "Running",
+                        "executedSuccessfully": "executed successfully",
+                        "failed": "failed",
+                        "feedbackUpdated": "Feedback updated",
+                        "updating": "Updating"
+                    }
+                },
+                "dropScreen": {
+                    "dropYourFilesHere": "Drop your files here"
+                },
+                "index": {
+                    "failedToUpload": "Failed to upload",
+                    "cancelledUploadOf": "Cancelled upload of",
+                    "couldNotReachServer": "Could not reach the server",
+                    "continuingChat": "Continuing previous chat"
+                },
+                "settings": {
+                    "settingsPanel": "Settings panel",
+                    "reset": "Reset",
+                    "cancel": "Cancel",
+                    "confirm": "Confirm"
+                }
+            },
+            "threadHistory": {
+                "sidebar": {
+                    "filters": {
+                        "FeedbackSelect": {
+                            "feedbackAll": "Feedback: All",
+                            "feedbackPositive": "Feedback: Positive",
+                            "feedbackNegative": "Feedback: Negative"
+                        },
+                        "SearchBar": {
+                            "search": "Search"
+                        }
+                    },
+                    "DeleteThreadButton": {
+                        "confirmMessage": "This will delete the thread as well as it's messages and elements.",
+                        "cancel": "Cancel",
+                        "confirm": "Confirm",
+                        "deletingChat": "Deleting chat",
+                        "chatDeleted": "Chat deleted"
+                    },
+                    "index": {
+                        "pastChats": "Past Chats"
+                    },
+                    "ThreadList": {
+                        "empty": "Empty...",
+                        "today": "Today",
+                        "yesterday": "Yesterday",
+                        "previous7days": "Previous 7 days",
+                        "previous30days": "Previous 30 days"
+                    },
+                    "TriggerButton": {
+                        "closeSidebar": "Close sidebar",
+                        "openSidebar": "Open sidebar"
+                    }
+                },
+                "Thread": {
+                    "backToChat": "Go back to chat",
+                    "chatCreatedOn": "This chat was created on"
+                }
+            },
+            "header": {
+                "chat": "Chat",
+                "readme": "Readme"
+            }
+        }
+    },
+    "hooks": {
+        "useLLMProviders": {
+            "failedToFetchProviders": "Failed to fetch providers:"
+        }
+    },
+    "pages": {
+        "Design": {},
+        "Env": {
+            "savedSuccessfully": "Saved successfully",
+            "requiredApiKeys": "Required API Keys",
+            "requiredApiKeysInfo": "To use this app, the following API keys are required. The keys are stored on your device's local storage."
+        },
+        "Page": {
+            "notPartOfProject": "You are not part of this project."
+        },
+        "ResumeButton": {
+            "resumeChat": "Resume Chat"
+        }
+    }
+}

.chainlit/translations/zh-CN.json

@@ -0,0 +1,229 @@
+{
+    "components": {
+        "atoms": {
+            "buttons": {
+                "userButton": {
+                    "menu": {
+                        "settings": "\u8bbe\u7f6e",
+                        "settingsKey": "S",
+                        "APIKeys": "API \u5bc6\u94a5",
+                        "logout": "\u767b\u51fa"
+                    }
+                }
+            }
+        },
+        "molecules": {
+            "newChatButton": {
+                "newChat": "\u65b0\u5efa\u5bf9\u8bdd"
+            },
+            "tasklist": {
+                "TaskList": {
+                    "title": "\ud83d\uddd2\ufe0f \u4efb\u52a1\u5217\u8868",
+                    "loading": "\u52a0\u8f7d\u4e2d...",
+                    "error": "\u53d1\u751f\u9519\u8bef"
+                }
+            },
+            "attachments": {
+                "cancelUpload": "\u53d6\u6d88\u4e0a\u4f20",
+                "removeAttachment": "\u79fb\u9664\u9644\u4ef6"
+            },
+            "newChatDialog": {
+                "createNewChat": "\u521b\u5efa\u65b0\u5bf9\u8bdd\uff1f",
+                "clearChat": "\u8fd9\u5c06\u6e05\u9664\u5f53\u524d\u6d88\u606f\u5e76\u5f00\u59cb\u65b0\u7684\u5bf9\u8bdd\u3002",
+                "cancel": "\u53d6\u6d88",
+                "confirm": "\u786e\u8ba4"
+            },
+            "settingsModal": {
+                "settings": "\u8bbe\u7f6e",
+                "expandMessages": "\u5c55\u5f00\u6d88\u606f",
+                "hideChainOfThought": "\u9690\u85cf\u601d\u8003\u94fe",
+                "darkMode": "\u6697\u8272\u6a21\u5f0f"
+            },
+            "detailsButton": {
+                "using": "\u4f7f\u7528",
+                "used": "\u5df2\u7528"
+            },
+            "auth": {
+                "authLogin": {
+                    "title": "\u767b\u5f55\u4ee5\u8bbf\u95ee\u5e94\u7528\u3002",
+                    "form": {
+                        "email": "\u7535\u5b50\u90ae\u7bb1\u5730\u5740",
+                        "password": "\u5bc6\u7801",
+                        "noAccount": "\u6ca1\u6709\u8d26\u6237\uff1f",
+                        "alreadyHaveAccount": "\u5df2\u6709\u8d26\u6237\uff1f",
+                        "signup": "\u6ce8\u518c",
+                        "signin": "\u767b\u5f55",
+                        "or": "\u6216\u8005",
+                        "continue": "\u7ee7\u7eed",
+                        "forgotPassword": "\u5fd8\u8bb0\u5bc6\u7801\uff1f",
+                        "passwordMustContain": "\u60a8\u7684\u5bc6\u7801\u5fc5\u987b\u5305\u542b\uff1a",
+                        "emailRequired": "\u7535\u5b50\u90ae\u7bb1\u662f\u5fc5\u586b\u9879",
+                        "passwordRequired": "\u5bc6\u7801\u662f\u5fc5\u586b\u9879"
+                    },
+                    "error": {
+                        "default": "\u65e0\u6cd5\u767b\u5f55\u3002",
+                        "signin": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "oauthsignin": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "redirect_uri_mismatch": "\u91cd\u5b9a\u5411URI\u4e0eOAuth\u5e94\u7528\u914d\u7f6e\u4e0d\u5339\u914d\u3002",
+                        "oauthcallbackerror": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "oauthcreateaccount": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "emailcreateaccount": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "callback": "\u5c1d\u8bd5\u4f7f\u7528\u4e0d\u540c\u7684\u8d26\u6237\u767b\u5f55\u3002",
+                        "oauthaccountnotlinked": "\u4e3a\u4e86\u9a8c\u8bc1\u60a8\u7684\u8eab\u4efd\uff0c\u8bf7\u4f7f\u7528\u6700\u521d\u4f7f\u7528\u7684\u540c\u4e00\u8d26\u6237\u767b\u5f55\u3002",
+                        "emailsignin": "\u65e0\u6cd5\u53d1\u9001\u90ae\u4ef6\u3002",
+                        "emailverify": "\u8bf7\u9a8c\u8bc1\u60a8\u7684\u7535\u5b50\u90ae\u4ef6\uff0c\u5df2\u53d1\u9001\u4e00\u5c01\u65b0\u90ae\u4ef6\u3002",
+                        "credentialssignin": "\u767b\u5f55\u5931\u8d25\u3002\u8bf7\u68c0\u67e5\u60a8\u63d0\u4f9b\u7684\u8be6\u7ec6\u4fe1\u606f\u662f\u5426\u6b63\u786e\u3002",
+                        "sessionrequired": "\u8bf7\u767b\u5f55\u4ee5\u8bbf\u95ee\u6b64\u9875\u9762\u3002"
+                    }
+                },
+                "authVerifyEmail": {
+                    "almostThere": "\u60a8\u5feb\u6210\u529f\u4e86\uff01\u6211\u4eec\u5df2\u5411 ",
+                    "verifyEmailLink": "\u8bf7\u5355\u51fb\u8be5\u90ae\u4ef6\u4e2d\u7684\u94fe\u63a5\u4ee5\u5b8c\u6210\u6ce8\u518c\u3002",
+                    "didNotReceive": "\u6ca1\u627e\u5230\u90ae\u4ef6\uff1f",
+                    "resendEmail": "\u91cd\u65b0\u53d1\u9001\u90ae\u4ef6",
+                    "goBack": "\u8fd4\u56de",
+                    "emailSent": "\u90ae\u4ef6\u5df2\u6210\u529f\u53d1\u9001\u3002",
+                    "verifyEmail": "\u9a8c\u8bc1\u60a8\u7684\u7535\u5b50\u90ae\u4ef6\u5730\u5740"
+                },
+                "providerButton": {
+                    "continue": "\u4f7f\u7528{{provider}}\u7ee7\u7eed",
+                    "signup": "\u4f7f\u7528{{provider}}\u6ce8\u518c"
+                },
+                "authResetPassword": {
+                    "newPasswordRequired": "\u65b0\u5bc6\u7801\u662f\u5fc5\u586b\u9879",
+                    "passwordsMustMatch": "\u5bc6\u7801\u5fc5\u987b\u4e00\u81f4",
+                    "confirmPasswordRequired": "\u786e\u8ba4\u5bc6\u7801\u662f\u5fc5\u586b\u9879",
+                    "newPassword": "\u65b0\u5bc6\u7801",
+                    "confirmPassword": "\u786e\u8ba4\u5bc6\u7801",
+                    "resetPassword": "\u91cd\u7f6e\u5bc6\u7801"
+                },
+                "authForgotPassword": {
+                    "email": "\u7535\u5b50\u90ae\u7bb1\u5730\u5740",
+                    "emailRequired": "\u7535\u5b50\u90ae\u7bb1\u662f\u5fc5\u586b\u9879",
+                    "emailSent": "\u8bf7\u68c0\u67e5\u7535\u5b50\u90ae\u7bb1{{email}}\u4ee5\u83b7\u53d6\u91cd\u7f6e\u5bc6\u7801\u7684\u6307\u793a\u3002",
+                    "enterEmail": "\u8bf7\u8f93\u5165\u60a8\u7684\u7535\u5b50\u90ae\u7bb1\u5730\u5740\uff0c\u6211\u4eec\u5c06\u53d1\u9001\u91cd\u7f6e\u5bc6\u7801\u7684\u6307\u793a\u3002",
+                    "resendEmail": "\u91cd\u65b0\u53d1\u9001\u90ae\u4ef6",
+                    "continue": "\u7ee7\u7eed",
+                    "goBack": "\u8fd4\u56de"
+                }
+            }
+        },
+        "organisms": {
+            "chat": {
+                "history": {
+                    "index": {
+                        "showHistory": "\u663e\u793a\u5386\u53f2",
+                        "lastInputs": "\u6700\u540e\u8f93\u5165",
+                        "noInputs": "\u5982\u6b64\u7a7a\u65f7...",
+                        "loading": "\u52a0\u8f7d\u4e2d..."
+                    }
+                },
+                "inputBox": {
+                    "input": {
+                        "placeholder": "\u5728\u8fd9\u91cc\u8f93\u5165\u60a8\u7684\u6d88\u606f..."
+                    },
+                    "speechButton": {
+                        "start": "\u5f00\u59cb\u5f55\u97f3",
+                        "stop": "\u505c\u6b62\u5f55\u97f3"
+                    },
+                    "SubmitButton": {
+                        "sendMessage": "\u53d1\u9001\u6d88\u606f",
+                        "stopTask": "\u505c\u6b62\u4efb\u52a1"
+                    },
+                    "UploadButton": {
+                        "attachFiles": "\u9644\u52a0\u6587\u4ef6"
+                    },
+                    "waterMark": {
+                        "text": "\u4f7f\u7528"
+                    }
+                },
+                "Messages": {
+                    "index": {
+                        "running": "\u8fd0\u884c\u4e2d",
+                        "executedSuccessfully": "\u6267\u884c\u6210\u529f",
+                        "failed": "\u5931\u8d25",
+                        "feedbackUpdated": "\u53cd\u9988\u66f4\u65b0",
+                        "updating": "\u6b63\u5728\u66f4\u65b0"
+                    }
+                },
+                "dropScreen": {
+                    "dropYourFilesHere": "\u5728\u8fd9\u91cc\u62d6\u653e\u60a8\u7684\u6587\u4ef6"
+                },
+                "index": {
+                    "failedToUpload": "\u4e0a\u4f20\u5931\u8d25",
+                    "cancelledUploadOf": "\u53d6\u6d88\u4e0a\u4f20",
+                    "couldNotReachServer": "\u65e0\u6cd5\u8fde\u63a5\u5230\u670d\u52a1\u5668",
+                    "continuingChat": "\u7ee7\u7eed\u4e4b\u524d\u7684\u5bf9\u8bdd"
+                },
+                "settings": {
+                    "settingsPanel": "\u8bbe\u7f6e\u9762\u677f",
+                    "reset": "\u91cd\u7f6e",
+                    "cancel": "\u53d6\u6d88",
+                    "confirm": "\u786e\u8ba4"
+                }
+            },
+            "threadHistory": {
+                "sidebar": {
+                    "filters": {
+                        "FeedbackSelect": {
+                            "feedbackAll": "\u53cd\u9988\uff1a\u5168\u90e8",
+                            "feedbackPositive": "\u53cd\u9988\uff1a\u6b63\u9762",
+                            "feedbackNegative": "\u53cd\u9988\uff1a\u8d1f\u9762"
+                        },
+                        "SearchBar": {
+                            "search": "\u641c\u7d22"
+                        }
+                    },
+                    "DeleteThreadButton": {
+                        "confirmMessage": "\u8fd9\u5c06\u5220\u9664\u7ebf\u7a0b\u53ca\u5176\u6d88\u606f\u548c\u5143\u7d20\u3002",
+                        "cancel": "\u53d6\u6d88",
+                        "confirm": "\u786e\u8ba4",
+                        "deletingChat": "\u5220\u9664\u5bf9\u8bdd",
+                        "chatDeleted": "\u5bf9\u8bdd\u5df2\u5220\u9664"
+                    },
+                    "index": {
+                        "pastChats": "\u8fc7\u5f80\u5bf9\u8bdd"
+                    },
+                    "ThreadList": {
+                        "empty": "\u7a7a\u7684...",
+                        "today": "\u4eca\u5929",
+                        "yesterday": "\u6628\u5929",
+                        "previous7days": "\u524d7\u5929",
+                        "previous30days": "\u524d30\u5929"
+                    },
+                    "TriggerButton": {
+                        "closeSidebar": "\u5173\u95ed\u4fa7\u8fb9\u680f",
+                        "openSidebar": "\u6253\u5f00\u4fa7\u8fb9\u680f"
+                    }
+                },
+                "Thread": {
+                    "backToChat": "\u8fd4\u56de\u5bf9\u8bdd",
+                    "chatCreatedOn": "\u6b64\u5bf9\u8bdd\u521b\u5efa\u4e8e"
+                }
+            },
+            "header": {
+                "chat": "\u5bf9\u8bdd",
+                "readme": "\u8bf4\u660e"
+            }
+        }
+    },
+    "hooks": {
+        "useLLMProviders": {
+            "failedToFetchProviders": "\u83b7\u53d6\u63d0\u4f9b\u8005\u5931\u8d25:"
+        }
+    },
+    "pages": {
+        "Design": {},
+        "Env": {
+            "savedSuccessfully": "\u4fdd\u5b58\u6210\u529f",
+            "requiredApiKeys": "\u5fc5\u9700\u7684API\u5bc6\u94a5",
+            "requiredApiKeysInfo": "\u8981\u4f7f\u7528\u6b64\u5e94\u7528\uff0c\u9700\u8981\u4ee5\u4e0bAPI\u5bc6\u94a5\u3002\u8fd9\u4e9b\u5bc6\u94a5\u5b58\u50a8\u5728\u60a8\u7684\u8bbe\u5907\u672c\u5730\u5b58\u50a8\u4e2d\u3002"
+        },
+        "Page": {
+            "notPartOfProject": "\u60a8\u4e0d\u662f\u6b64\u9879\u76ee\u7684\u4e00\u90e8\u5206\u3002"
+        },
+        "ResumeButton": {
+            "resumeChat": "\u6062\u590d\u5bf9\u8bdd"
+        }
+    }
+}

app.py

@@ -1,28 +1,17 @@
-import os
+import os, tempfile
 from typing import List
 from chainlit.types import AskFileResponse
 from aimakerspace.text_utils import CharacterTextSplitter, TextFileLoader
-from aimakerspace.openai_utils.prompts import (
-    UserRolePrompt,
-    SystemRolePrompt,
-    AssistantRolePrompt,
-)
-from aimakerspace.openai_utils.embedding import EmbeddingModel
+from aimakerspace.openai_utils.prompts import UserRolePrompt, SystemRolePrompt
 from aimakerspace.vectordatabase import VectorDatabase
 from aimakerspace.openai_utils.chatmodel import ChatOpenAI
 import chainlit as cl
+from PyPDF2 import PdfReader
 
-system_template = """\
-Use the following context to answer a users question. If you cannot find the answer in the context, say you don't know the answer."""
+system_template = "Use the following context to answer a users question. If you cannot find the answer in the context, say you don't know the answer."
 system_role_prompt = SystemRolePrompt(system_template)
 
-user_prompt_template = """\
-Context:
-{context}
-
-Question:
-{question}
-"""
+user_prompt_template = "Context:\n{context}\n\nQuestion:\n{question}"
 user_role_prompt = UserRolePrompt(user_prompt_template)
 
 class RetrievalAugmentedQAPipeline:
@@ -32,13 +21,8 @@ class RetrievalAugmentedQAPipeline:
 
     async def arun_pipeline(self, user_query: str):
         context_list = self.vector_db_retriever.search_by_text(user_query, k=4)
-
-        context_prompt = ""
-        for context in context_list:
-            context_prompt += context[0] + "\n"
-
+        context_prompt = "\n".join([context[0] for context in context_list])
         formatted_system_prompt = system_role_prompt.create_message()
-
         formatted_user_prompt = user_role_prompt.create_message(question=user_query, context=context_prompt)
 
         async def generate_response():
@@ -49,70 +33,56 @@ class RetrievalAugmentedQAPipeline:
 
 text_splitter = CharacterTextSplitter()
 
-
-def process_text_file(file: AskFileResponse):
-    import tempfile
-
-    with tempfile.NamedTemporaryFile(mode="w", delete=False, suffix=".txt") as temp_file:
+def process_file(file: AskFileResponse):
+    with tempfile.NamedTemporaryFile(mode="wb", delete=False, suffix=file.name) as temp_file:
+        temp_file.write(file.content)
         temp_file_path = temp_file.name
 
-    with open(temp_file_path, "wb") as f:
-        f.write(file.content)
+    if file.type == "text/plain":
+        text_loader = TextFileLoader(temp_file_path)
+        documents = text_loader.load_documents()
+    elif file.type == "application/pdf":
+        pdf_reader = PdfReader(temp_file_path)
+        documents = [page.extract_text() for page in pdf_reader.pages]
+    else:
+        raise ValueError(f"Unsupported file type: {file.type}")
 
-    text_loader = TextFileLoader(temp_file_path)
-    documents = text_loader.load_documents()
     texts = text_splitter.split_texts(documents)
+    os.unlink(temp_file_path)
     return texts
 
-
 @cl.on_chat_start
 async def on_chat_start():
     files = None
-
-    # Wait for the user to upload a file
     while files == None:
         files = await cl.AskFileMessage(
-            content="Please upload a Text File file to begin!",
-            accept=["text/plain"],
-            max_size_mb=2,
+            content="Please upload a Text or PDF file to begin!",
+            accept=["text/plain", "application/pdf"],
+            max_size_mb=20,
             timeout=180,
         ).send()
 
     file = files[0]
-
-    msg = cl.Message(
-        content=f"Processing `{file.name}`...", disable_human_feedback=True
-    )
+    msg = cl.Message(content=f"Processing `{file.name}`...", disable_human_feedback=True)
     await msg.send()
 
-    # load the file
-    texts = process_text_file(file)
-
+    texts = process_file(file)
     print(f"Processing {len(texts)} text chunks")
 
-    # Create a dict vector store
     vector_db = VectorDatabase()
     vector_db = await vector_db.abuild_from_list(texts)
     
     chat_openai = ChatOpenAI()
-
-    # Create a chain
-    retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(
-        vector_db_retriever=vector_db,
-        llm=chat_openai
-    )
+    retrieval_augmented_qa_pipeline = RetrievalAugmentedQAPipeline(vector_db_retriever=vector_db, llm=chat_openai)
     
-    # Let the user know that the system is ready
     msg.content = f"Processing `{file.name}` done. You can now ask questions!"
     await msg.update()
 
     cl.user_session.set("chain", retrieval_augmented_qa_pipeline)
 
-
 @cl.on_message
 async def main(message):
     chain = cl.user_session.get("chain")
-
     msg = cl.Message(content="")
     result = await chain.arun_pipeline(message.content)