Added Files

.gitignore

@@ -0,0 +1,160 @@
+# Byte-compiled / optimized / DLL files
+__pycache__/
+*.py[cod]
+*$py.class
+
+# C extensions
+*.so
+
+# Distribution / packaging
+.Python
+build/
+develop-eggs/
+dist/
+downloads/
+eggs/
+.eggs/
+lib/
+lib64/
+parts/
+sdist/
+var/
+wheels/
+share/python-wheels/
+*.egg-info/
+.installed.cfg
+*.egg
+MANIFEST
+
+# PyInstaller
+#  Usually these files are written by a python script from a template
+#  before PyInstaller builds the exe, so as to inject date/other infos into it.
+*.manifest
+*.spec
+
+# Installer logs
+pip-log.txt
+pip-delete-this-directory.txt
+
+# Unit test / coverage reports
+htmlcov/
+.tox/
+.nox/
+.coverage
+.coverage.*
+.cache
+nosetests.xml
+coverage.xml
+*.cover
+*.py,cover
+.hypothesis/
+.pytest_cache/
+cover/
+
+# Translations
+*.mo
+*.pot
+
+# Django stuff:
+*.log
+local_settings.py
+db.sqlite3
+db.sqlite3-journal
+
+# Flask stuff:
+instance/
+.webassets-cache
+
+# Scrapy stuff:
+.scrapy
+
+# Sphinx documentation
+docs/_build/
+
+# PyBuilder
+.pybuilder/
+target/
+
+# Jupyter Notebook
+.ipynb_checkpoints
+
+# IPython
+profile_default/
+ipython_config.py
+
+# pyenv
+#   For a library or package, you might want to ignore these files since the code is
+#   intended to run in multiple environments; otherwise, check them in:
+# .python-version
+
+# pipenv
+#   According to pypa/pipenv#598, it is recommended to include Pipfile.lock in version control.
+#   However, in case of collaboration, if having platform-specific dependencies or dependencies
+#   having no cross-platform support, pipenv may install dependencies that don't work, or not
+#   install all needed dependencies.
+#Pipfile.lock
+
+# poetry
+#   Similar to Pipfile.lock, it is generally recommended to include poetry.lock in version control.
+#   This is especially recommended for binary packages to ensure reproducibility, and is more
+#   commonly ignored for libraries.
+#   https://python-poetry.org/docs/basic-usage/#commit-your-poetrylock-file-to-version-control
+#poetry.lock
+
+# pdm
+#   Similar to Pipfile.lock, it is generally recommended to include pdm.lock in version control.
+#pdm.lock
+#   pdm stores project-wide configurations in .pdm.toml, but it is recommended to not include it
+#   in version control.
+#   https://pdm.fming.dev/#use-with-ide
+.pdm.toml
+
+# PEP 582; used by e.g. github.com/David-OConnor/pyflow and github.com/pdm-project/pdm
+__pypackages__/
+
+# Celery stuff
+celerybeat-schedule
+celerybeat.pid
+
+# SageMath parsed files
+*.sage.py
+
+# Environments
+.env
+.venv
+env/
+venv/
+ENV/
+env.bak/
+venv.bak/
+
+# Spyder project settings
+.spyderproject
+.spyproject
+
+# Rope project settings
+.ropeproject
+
+# mkdocs documentation
+/site
+
+# mypy
+.mypy_cache/
+.dmypy.json
+dmypy.json
+
+# Pyre type checker
+.pyre/
+
+# pytype static type analyzer
+.pytype/
+
+# Cython debug symbols
+cython_debug/
+
+# PyCharm
+#  JetBrains specific template is maintained in a separate JetBrains.gitignore that can
+#  be found at https://github.com/github/gitignore/blob/main/Global/JetBrains.gitignore
+#  and can be added to the global gitignore or merged into this file.  For a more nuclear
+#  option (not recommended) you can uncomment the following to ignore the entire idea folder.
+#.idea/

README.md

@@ -1,12 +0,0 @@
----
-title: Legal Assitr Poc
-emoji: 🚀
-colorFrom: purple
-colorTo: purple
-sdk: gradio
-sdk_version: 3.44.4
-app_file: app.py
-pinned: false
----
-
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,175 @@
+from datetime import datetime
+import gradio as gr
+import json
+import time
+from legal_assitr_prototype.engine.engine import SYSTEM_PROMPT, generate, generate_user_prompt
+
+from legal_assitr_prototype.templates.nda.nda_form import NDA_TEMPLATE_PROMPT
+from legal_assitr_prototype.templates.templates_sidebar import templates_sidebar
+
+DESCRIPTION = """
+# Legal Assitr PoC ⚖️ - This is a PoC for Legal Assitr
+### Legal Assist is an AI tool designed to assist users in creating legal documents by leveraging artificial intelligence (AI) technology.
+
+`retro56/zs-writer` and `retro56/legalwriter-7b-instruct`(best) perform the best but try out different models to see how they react to the same prompt.
+
+### Built by [Ankur Debnath](https://github.com/r3tr056) with 💙
+"""
+
+MAX_MAX_NEW_TOKENS = 10
+DEFAULT_MAX_NEW_TOKENS = 10
+
+curr_date = datetime.now().strftime("%m/%d/%Y")
+shared = dict()
+
+def main():
+
+    def complete(system_prompt, effective_date, disclosing_party, receiving_party, confidential_info, obligations, term, breach, city):
+
+        form_prompt = NDA_TEMPLATE_PROMPT.format(
+            effective_date=effective_date,
+            disclosing_party=disclosing_party,
+            receiving_party=receiving_party,
+            confidential_info=confidential_info,
+            obligations=obligations,
+            term=term,
+            breach=breach,
+            city=city
+        )
+        
+        result = generate(system_prompt=system_prompt, user_prompt=form_prompt)
+        result = f"""{result}"""
+        return result
+
+    with gr.Blocks(theme='soft') as demo:
+        gr.Markdown(DESCRIPTION)
+        with gr.Row():
+
+            with gr.Column(scale=2):
+                with gr.Blocks() as nda_form:
+                    with gr.Group():
+                        effective_date = gr.Textbox(
+                            label="Effective Date",
+                            placeholder=curr_date,
+                            value=curr_date
+                        )
+                        city = gr.Textbox(
+                            label="City"
+                        )
+                        purpose = gr.Textbox(
+                            label="Purpose of the NDA",
+                            placeholder="Enter the purpose of the NDA"
+                        )
+                        parties = gr.Textbox(
+                            label="Parties Involved",
+                            placeholder="Parties Involved..."
+                        )
+                        disclosing_party = gr.Textbox(
+                            label="Disclosing Party",
+                            placeholder="Disclosing Party with Address and Contact..."
+                        )
+                        receiving_party = gr.Textbox(
+                            label="Receiving Party",
+                            placeholder="Receiving Party with Address And Contact..."
+                        )
+                        obligations = gr.Textbox(
+                            label="Obligations",
+                            placeholder="Obligations"
+                        )
+                        mutual_or_not = gr.Textbox(
+                            label="Mutual or One Way NDA",
+                            placeholder="Whether the NDA is mutual or One-Way..."
+                        )
+                        confidential_info = gr.Textbox(
+                            label="Confidential Information",
+                            placeholder="Explain the Confidential Information"
+                        )
+                        term = gr.Textbox(
+                            label="Term",
+                            placeholder="How long the Confidentiality lasts",
+                        )
+                        breach = gr.Textbox(
+                            label="Consequences of breach",
+                            placeholder="Consequences of breach"
+                        )
+
+                with gr.Group():
+                    with gr.Accordion(label="AI Settings", open=False):
+                        system_prompt = gr.Textbox(
+                            label='System Prompt',
+                            value=SYSTEM_PROMPT,
+                            lines=5
+                        )
+
+                        max_new_tokens = gr.Slider(
+                            label='Max new tokens',
+                            minimum=1,
+                            maximum=MAX_MAX_NEW_TOKENS,
+                            step=1,
+                            value=DEFAULT_MAX_NEW_TOKENS,
+                        )
+                        temp = gr.Slider(
+                            label='Temperature',
+                            minimum=0.1,
+                            maximum=4.0,
+                            step=0.1,
+                            value=1.0,
+                        )
+                        top_p = gr.Slider(
+                            label='Top-p (nucleus sampling)',
+                            minimum=0.05,
+                            maximum=1.0,
+                            step=0.05,
+                            value=0.95,
+                        )
+                        top_k = gr.Slider(
+                            label='Top-k',
+                            minimum=1,
+                            maximum=1000,
+                            step=1,
+                            value=50,
+                        )
+
+            with gr.Column(scale=4):
+                with gr.Row():
+                    with gr.Tab("Raw"):
+                        raw_editor_textbox = gr.Textbox(
+                            label="Generated Document...",
+                            lines=35,
+                            elem_id='textbox-default',
+                            elem_classes=['textbox_default_output', 'add_scrollbar'],
+                            show_copy_button=True
+                        )
+                    with gr.Tab("Prettified"):
+                        md_render_btn = gr.Button('Render')
+                        md_editor_box = gr.Markdown()
+
+                    with gr.Tab("Generated Tokens"):
+                        shared["tokens-default"] = gr.Textbox(
+                            show_copy_button=True,
+                            lines=35,
+                            label="Tokens",
+                            elem_classes=['textbox_tokens', 'add_scrollbar', 'monospace']
+                        )
+                with gr.Row():
+                    generate_btn = gr.Button(value="Generate Content", variant='primary')
+                    render_md = gr.Button(value="Prettify", variant='secondary')
+        
+        generate_btn.click(fn=complete, inputs=[
+            system_prompt,
+            effective_date,
+            disclosing_party,
+            receiving_party,
+            confidential_info,
+            obligations,
+            term,
+            breach,
+            city
+        ], outputs=[raw_editor_textbox])
+        
+
+    demo.launch()
+
+
+if __name__ == "__main__":
+    main()

legal_assitr_prototype/engine/engine.py

@@ -0,0 +1,63 @@
+
+import os
+import openai
+
+openai.api_key = "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"
+
+GPT_MODEL = "gpt-3.5-turbo"
+
+SYSTEM_PROMPT = """
+You are 'LegalAssitr', A Top Quality Lawyer in Indian Law.
+Your skills are : 
+
+1. Legal Writing: Have exceptional writing skills with a deep understanding of legal language, terminology, and principles. They should be able to create clear, concise, and legally accurate documents.
+2. Contract Drafting and Review: Proficiency in drafting, reviewing, and negotiating contracts is essential. They must create legally sound agreements that protect the company's interests while ensuring clarity and enforceability.
+3. Legal Expertise: Have a strong understanding of various areas of law relevant to the company's operations, such as contract law, corporate law, employment law, intellectual property law, and regulatory compliance.
+4. Attention to Detail: You must pay close attention to every word, punctuation mark, and clause to prevent ambiguities or errors that could lead to legal disputes.
+5. Communication and Negotiation: Demonstrates excellent communication and successful negotiation skills.
+6. Document Review: Excels in review and analyze legal documents, contracts, and agreements for accuracy and compliance with legal requirements.
+7. Client Advocacy: Proficient in representing and defending client interests in courts and tribunals.
+8. Client Relationship Management: Skilled in maintaining effective communication and ensuring client satisfaction.
+
+You generate output STRICTLY in MARKDOWN fromat
+Please
+
+However, it's essential to underline the importance of providing accurate information. Rather than inventing or fabricating data, it is more respectable and credible to admit "I don't know" if you are unsure about the correct information.
+"""
+
+SAMPLE_USER_PROMPT = """
+Can you draft a lease agreement for a commercial property in New York that includes provisions on rent, security deposits, and maintenance obligations? Please cite relevant New York statutes and case law that apply to commercial leases.
+"""
+
+def generate(
+        system_prompt:str,
+        user_prompt:str=SAMPLE_USER_PROMPT,
+        temperature:float =0.9,
+        max_new_tokens:int=1000,
+        top_p:float=0.95,
+        repetition_penalty:float=1.0
+    ):
+    temperature = float(temperature)
+    if temperature < 1e-2:
+        temperature = 1e-2
+
+    top_p = float(top_p)
+    repetition_penalty = float(repetition_penalty)
+    
+    completion = openai.ChatCompletion.create(
+        model=GPT_MODEL,
+        messages=[
+            {"role": "system", "content": system_prompt},
+            {"role": "user", "content": user_prompt}
+        ],
+        temperature=temperature,
+        max_tokens=max_new_tokens,
+        frequency_penalty=repetition_penalty,
+        top_p=top_p,
+    )
+    return completion.choices[0].message["content"]
+
+def generate_user_prompt(form_data):
+    pass
+    
+

legal_assitr_prototype/templates/nda/nda_form.py

@@ -0,0 +1,18 @@
+from datetime import datetime
+
+NDA_TEMPLATE_PROMPT = """
+Craft a Non-Disclosure Agreement effective from {effective_date}, The NDA should cover the following key points:
+
+1. In the context of this agreement, {disclosing_party} is referred to as the the "Disclosing Party" and {receiving_party} is referred to as the "Receiving Party" respectively.
+2. The definition of 'Confidential Information' is {confidential_info} in the context of this agreement"
+3. Obligations of the Receiving Party, including non-disclosure and non-use of the Confidential Information and {obligations}.
+4. The duration of the NDA {term}.
+5. Permitted disclosures, if any, such as to employees or contractors who have a need to know the information for a specific purpose.
+6. Any exceptions to confidentiality, such as information that is already in the public domain or required to be disclosed by law.
+7. Consequences of breach, including remedies, injunctive relief, and any monetary damages and or {breach}.
+8. Governing law and jurisdiction for NDA in the city or place of {city}
+9. Any other standard clauses or legal language necessary to create a comprehensive and legally binding NDA.
+
+Please provide a draft NDA that adheres to these requirements, and ensure it's legally sound and precise. Thank you.
+"""
+

legal_assitr_prototype/templates/templates_sidebar.py

@@ -0,0 +1,14 @@
+import gradio as gr
+
+selected_doc_type = None
+
+def templates_sidebar():
+    with gr.Blocks() as sidebar:
+        with gr.Column():
+                with gr.Group():
+                    buy_button = gr.Button(
+                        icon="./assets/icons/buy.png",
+                    )
+                    contract_button = gr.Button(icon="./assets/icons/contract.png")
+                    terms_button = gr.Button(icon="./assets/icons/terms.png")
+    

pyproject.toml

@@ -0,0 +1,15 @@
+[tool.poetry]
+name = "legal-assitr-prototype"
+version = "0.1.0"
+description = ""
+authors = ["hopeJRD <jordanh0pe@pm.me>"]
+readme = "README.md"
+packages = [{include = "legal_assitr_prototype"}]
+
+[tool.poetry.dependencies]
+python = "^3.10"
+
+
+[build-system]
+requires = ["poetry-core"]
+build-backend = "poetry.core.masonry.api"