added history for generated text

app.py

@@ -6,7 +6,7 @@ import re
 import requests
 from typing import Dict
 from collections import defaultdict
-from datetime import date
+from datetime import date, datetime
 import gradio as gr
 from scipy.special import softmax
 import language_tool_python
@@ -373,6 +373,13 @@ def generate_article(
     return clean_text(article)
 
 
+history = []
+
+
+def get_history():
+    return history
+
+
 def humanize(
     text: str,
     model: str,
@@ -392,7 +399,11 @@ def humanize(
         length_penalty=length_penalty,
     )
     result = result + references
-    return format_and_correct_language_check(result)
+    corrected_text = format_and_correct_language_check(result)
+
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    history.append((f"Humanized Text | {timestamp}\nInput: {model}", corrected_text))
+    return corrected_text
 
 
 def update_visibility_api(model: str):
@@ -482,7 +493,10 @@ def generate_and_format(
         for url in url_content.keys():
             article += f"\n{url}"
 
-    return format_references(article)
+    reference_formatted = format_references(article)
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    history.append((f"Generated Text | {timestamp}\nInput: {topic}", reference_formatted))
+    return reference_formatted
 
 
 def create_interface():
@@ -681,26 +695,26 @@ def create_interface():
 
                     gr.Markdown("# Add Optional PDF Files with Information", elem_classes="text-center text-3xl mb-6")
                     pdf_file_input = gr.File(label="Upload PDF(s)", file_count="multiple", file_types=[".pdf"])
-
+                """
                 # NOTE: HIDE AI MODEL SELECTION
-                # with gr.Group():
-                #     gr.Markdown("## AI Model Configuration", elem_classes="text-xl mb-4")
-                #     ai_generator = gr.Dropdown(
-                #         choices=[
-                #             "OpenAI GPT 4",
-                #             "OpenAI GPT 4o",
-                #             "OpenAI GPT 4o Mini",
-                #             "Claude Sonnet 3.5",
-                #             "Gemini 1.5 Pro",
-                #             "LLaMA 3",
-                #         ],
-                #         value="OpenAI GPT 4o Mini",
-                #         label="AI Model",
-                #         elem_classes="input-highlight-pink",
-                #     )
-                # input_api = gr.Textbox(label="API Key", visible=False)
-                # ai_generator.change(update_visibility_api, ai_generator, input_api)
-
+                with gr.Group():
+                    gr.Markdown("## AI Model Configuration", elem_classes="text-xl mb-4")
+                    ai_generator = gr.Dropdown(
+                        choices=[
+                            "OpenAI GPT 4",
+                            "OpenAI GPT 4o",
+                            "OpenAI GPT 4o Mini",
+                            "Claude Sonnet 3.5",
+                            "Gemini 1.5 Pro",
+                            "LLaMA 3",
+                        ],
+                        value="OpenAI GPT 4o Mini",
+                        label="AI Model",
+                        elem_classes="input-highlight-pink",
+                    )
+                input_api = gr.Textbox(label="API Key", visible=False)
+                ai_generator.change(update_visibility_api, ai_generator, input_api)
+                """
                 generate_btn = gr.Button("Generate Article", variant="primary")
 
                 with gr.Accordion("Advanced Humanizer Settings", open=False):
@@ -728,23 +742,29 @@ def create_interface():
                         )
 
             with gr.Column(scale=3):
-                output_article = gr.Textbox(label="Generated Article", lines=20)
-                ai_comments = gr.Textbox(
-                    label="Add comments to help edit generated text", interactive=True, visible=False
-                )
-                regenerate_btn = gr.Button("Regenerate Article", variant="primary", visible=False)
-                ai_detector_dropdown = gr.Radio(
-                    choices=ai_check_options, label="Select AI Detector", value="Polygraf AI"
-                )
-                ai_check_btn = gr.Button("AI Check")
-
-                with gr.Accordion("AI Detection Results", open=True):
-                    ai_check_result = gr.Label(label="AI Check Result")
-                    highlighted_text = gr.HTML(label="Sentence Breakdown", visible=False)
-                humanize_btn = gr.Button("Humanize")
-                # humanized_output = gr.Textbox(label="Humanized Article", lines=20, elem_classes=["custom-textbox"])
-                humanized_output = gr.Markdown(label="Humanized Article", value="\n\n\n\n", render=True)
-                copy_to_input_btn = gr.Button("Copy to Input for AI Check")
+                with gr.Tab("Text Generator"):
+                    output_article = gr.Textbox(label="Generated Article", lines=20)
+                    ai_comments = gr.Textbox(
+                        label="Add comments to help edit generated text", interactive=True, visible=False
+                    )
+                    regenerate_btn = gr.Button("Regenerate Article", variant="primary", visible=False)
+                    ai_detector_dropdown = gr.Radio(
+                        choices=ai_check_options, label="Select AI Detector", value="Polygraf AI"
+                    )
+                    ai_check_btn = gr.Button("AI Check")
+
+                    with gr.Accordion("AI Detection Results", open=True):
+                        ai_check_result = gr.Label(label="AI Check Result")
+                        highlighted_text = gr.HTML(label="Sentence Breakdown", visible=False)
+                    humanize_btn = gr.Button("Humanize")
+                    # humanized_output = gr.Textbox(label="Humanized Article", lines=20, elem_classes=["custom-textbox"])
+                    humanized_output = gr.Markdown(label="Humanized Article", value="\n\n\n\n", render=True)
+                    # copy_to_input_btn = gr.Button("Copy to Input for AI Check")
+
+                with gr.Tab("History"):
+                    history_chat = gr.Chatbot(label="Generation History")
+                    refresh_button = gr.Button("Refresh History")
+                    refresh_button.click(get_history, outputs=history_chat)
 
         def regenerate_visible(text):
             if text:
@@ -856,12 +876,6 @@ def create_interface():
                 top_k_slider,
                 length_penalty_slider,
             ],
-            outputs=[humanized_output],
-        )
-
-        copy_to_input_btn.click(
-            fn=copy_to_input,
-            inputs=[humanized_output],
             outputs=[output_article],
         )