Create app.py

app.py

@@ -0,0 +1,114 @@
+from legal_empowerment.PlayGround import respond, reset_chat
+
+# For our main chatbot
+from legal_empowerment.OpenAIAssistant import respond as main_assistant_respond
+from legal_empowerment.OpenAIAssistant import reset_chat as main_assistant_reset_chat
+
+import gradio as gr
+import os 
+
+# Create a fake download csv file to test the download button from gradio 
+def create_draft_text():
+    draft_text = "This is a draft text file.\nFeel free to edit as needed."
+    file_name = "draft_text.txt"
+    file_path = os.path.join(os.getcwd(), file_name)
+    with open(file_path, "w") as file:
+        file.write(draft_text)
+    return file_path
+
+    
+main_markdown_text = """
+# Legal Empowerment Project Summary
+## Empowering Tennesseans with AI for Life and Death Planning
+
+This project aims to leverage generative AI to assist Tennesseans in planning for important life events and managing end-of-life processes. Our solution bridges the gap in accessibility and understanding of legal resources by providing an AI-powered chatbot that helps users with:
+
+1. **Advance Directives**: Assisting in planning for future medical decisions and filling out advance directive forms.
+2. **Wills**: Guiding users in creating wills to plan for the distribution of their assets.
+3. **Probate Management**: Offering guidance through the probate process after a loved one's passing.
+
+Our platform includes two main components:
+
+1. **Chatbot Tab**: A user-friendly AI assistant that answers questions and helps fill out advance directive forms, culminating in a downloadable document.
+2. **Playground Tab**: An experimental area where users can explore and test responses from various AI models, including OpenAI Completion, OpenAI Assistant, and Langchain Agent. This tab is designed for experimentation and does not support form filling.
+
+With this project, we aim to make the planning process more accessible and manageable for everyone, regardless of their legal knowledge.
+"""
+
+# Markdown for tab 1
+tab1_markdown = """
+Welcome to the **Final Chatbot**! This section features our AI-powered assistant (`gpt-4o`) designed to help you with:
+
+- **Advance Directives:** Get assistance with planning your future medical decisions and filling out advance directive forms.
+- **Wills:** Receive guidance on creating a will to manage the distribution of your assets.
+- **Probate Management:** Navigate the probate process with ease after a loved one’s passing.
+
+Simply ask your questions or follow the prompts, and our chatbot will guide you through each step. Once completed, you can download your personalized advance directive form directly from this tab.
+"""
+
+# Markdown for tab 2 - Playground
+tab2_markdown = """
+Welcome to the **Playground**! This area allows you to experiment with different AI models, using `gpt-4o`, including:
+
+- **OpenAI Completion** 
+- **OpenAI Assistant** 
+- **Langchain Agent** built in web lookup and RAG
+
+Use this space to understand the capabilities and nuances of each model. Please note that this tab is for experimentation only and does not support form filling or creating advance directive forms.
+"""
+
+# Defines a list of the available approaches
+approaches = ["Completion Model", "Langchain Agent RAG", "OpenAI Assistant"]
+with gr.Blocks() as demo:
+
+    gr.Markdown("#")
+    gr.Markdown(main_markdown_text)
+
+
+    with gr.Tabs():
+        # Tab 1: 
+        with gr.TabItem("Chatbot"):
+            gr.Markdown(tab1_markdown)
+
+            # Create a chatbot section
+            chatbot_history = gr.Chatbot()  # This will store the chat history
+            msg_textbox = gr.Textbox(placeholder="Type a message...")  # This is where the user types their message
+            reset_button = gr.Button("Clear Chat")  # Button to clear the chat history
+
+             # Define what happens when the user submits a message
+            msg_textbox.submit(main_assistant_respond, inputs=[msg_textbox, chatbot_history], outputs=[msg_textbox, chatbot_history])
+
+            # Define what happens when the reset button is clicked
+            reset_button.click(main_assistant_reset_chat,  outputs=[chatbot_history, msg_textbox])
+            
+            # Form download section 
+            gr.Markdown("## Download Advance Directive form")
+            download_button = gr.Button("Generate and Download")
+            download_file = gr.File()
+            download_button.click(fn=create_draft_text, inputs=[], outputs=download_file)
+            
+            
+        # Tab 2: Playground
+        with gr.TabItem("PlayGround"):
+            gr.Markdown(tab2_markdown)
+            
+            # The drop-down to select what model 
+            with gr.Row():
+                with gr.Column(scale=1):
+                    approach_dropdown = gr.Dropdown(choices=approaches, label="Select Approach") # Creates the dropdown for selecting an approach
+
+            # Create a chatbot section
+            chatbot_history = gr.Chatbot()  # This will store the chat history
+            msg_textbox = gr.Textbox(placeholder="Type a message...")  # This is where the user types their message
+            reset_button = gr.Button("Clear Chat")  # Button to clear the chat history
+
+            # Define what happens when the user submits a message
+            msg_textbox.submit(respond, inputs=[msg_textbox, approach_dropdown, chatbot_history], outputs=[msg_textbox, chatbot_history])
+
+            # Define what happens when the reset button is clicked
+            reset_button.click(reset_chat,inputs = [approach_dropdown],  outputs=[chatbot_history, msg_textbox])
+
+    gr.Markdown("### Thank you for using our Legal Empowerment Interface!")  # Closing message
+
+# Launch the interface
+demo.launch()