Update app.py

app.py

@@ -54,4 +54,21 @@ def main():
                 for line in block.lines:
                     for word in line.words:
                         text += word.value + " "
-                text += "\n
+                text += "\n"
+
+        # Prepare the input for the LLM
+        messages = [
+            {"role": "system", "content": "You are a helpful AI assistant."},
+            {"role": "user", "content": f"Extract the email, phone number, and location from the following text:\n{text}"}
+        ]
+
+        # Use the LLM to extract the details
+        output = pipe(messages, **generation_args)
+        generated_text = output[0]['generated_text']
+
+        # Display the extracted information
+        st.header('Extracted Information')
+        st.write(generated_text)
+
+if __name__ == '__main__':
+    main()