fixed shared history + add clear history button

app.py

@@ -377,11 +377,13 @@ def generate_article(
     return clean_text(article)
 
 
-history = []
+def get_history(history):
+    return history
 
 
-def get_history():
-    return history
+def clear_history():
+    # Return empty list for history state and display
+    return [], []
 
 
 def humanize(
@@ -391,6 +393,7 @@ def humanize(
     repetition_penalty: float = 1,
     top_k: int = 50,
     length_penalty: float = 1,
+    history=None,
 ) -> str:
     print("Humanizing text...")
     body, references = split_text_from_refs(text)
@@ -407,7 +410,7 @@ def humanize(
 
     timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
     history.append((f"Humanized Text | {timestamp}\nInput: {model}", corrected_text))
-    return corrected_text
+    return corrected_text, history
 
 
 def update_visibility_api(model: str):
@@ -443,6 +446,7 @@ def generate_and_format(
     include_sites,
     exclude_sites,
     pdf_file_input,
+    history=None,
     ai_model="OpenAI GPT 4o",
     api_key=None,
     generated_article: str = None,
@@ -500,7 +504,7 @@ def generate_and_format(
     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
+    return reference_formatted, history
 
 
 def create_interface():
@@ -512,6 +516,7 @@ def create_interface():
             .input-highlight-pink block_label {background-color: #008080}
             """,
     ) as demo:
+        history = gr.State([])
         today = date.today()
         # dd/mm/YY
         d1 = today.strftime("%d/%B/%Y")
@@ -766,6 +771,8 @@ def create_interface():
 
                 with gr.Tab("History"):
                     history_chat = gr.Chatbot(label="Generation History", height=1000)
+                    clear_history_btn = gr.Button("Clear History")
+                    clear_history_btn.click(clear_history, outputs=[history, history_chat])
                     """
                     # NOTE: REMOVED REFRESH BUTTON
                     refresh_button = gr.Button("Refresh History")
@@ -826,8 +833,9 @@ def create_interface():
                 include_sites,
                 exclude_sites,
                 pdf_file_input,
+                history,
             ],
-            outputs=[output_article],
+            outputs=[output_article, history],
         )
 
         regenerate_btn.click(
@@ -858,12 +866,13 @@ def create_interface():
                 day_to,
                 domains_to_include,
                 pdf_file_input,
+                history,
                 output_article,
                 include_sites,
                 exclude_sites,
                 ai_comments,
             ],
-            outputs=[output_article],
+            outputs=[output_article, history],
         )
 
         ai_check_btn.click(
@@ -881,13 +890,14 @@ def create_interface():
                 repetition_penalty_slider,
                 top_k_slider,
                 length_penalty_slider,
+                history,
             ],
-            outputs=[output_article],
+            outputs=[output_article, history],
         )
 
-        generate_btn.click(get_history, outputs=history_chat)
-        regenerate_btn.click(get_history, outputs=history_chat)
-        humanize_btn.click(get_history, outputs=history_chat)
+        generate_btn.click(get_history, inputs=[history], outputs=[history_chat])
+        regenerate_btn.click(get_history, inputs=[history], outputs=[history_chat])
+        humanize_btn.click(get_history, inputs=[history], outputs=[history_chat])
 
     return demo