add regeneration of ai text based on comments

app.py

@@ -74,6 +74,29 @@ def generate_prompt(settings: Dict[str, str]) -> str:
     return prompt
 
 
+def regenerate_prompt(settings: Dict[str, str]) -> str:
+    """Edit the given text based on user settings and comments."""
+    prompt = f"""
+    "{settings['generated_article']}"
+
+    Edit the given text based on user comments.
+    
+    Comments:
+    - {settings['user_comments']}
+
+    Additional requirements:
+    - Include {settings['num_examples']} relevant examples or case studies
+    - Incorporate data or statistics from {', '.join(settings['references'])}
+    - End with a {settings['conclusion_type']} conclusion
+    - Add a "References" section at the end with at least 3 credible sources, formatted as [1], [2], etc.
+    - Do not make any headline, title bold.
+    
+    Ensure proper paragraph breaks for better readability.
+    Avoid any references to artificial intelligence, language models, or the fact that this is generated by an AI, and do not mention something like here is the article etc.
+    """
+    return prompt
+
+
 def generate_article(
     topic: str,
     keywords: str,
@@ -89,6 +112,8 @@ def generate_article(
     conclusion_type: str,
     ai_model: str,
     api_key: str = None,
+    generated_article: str = None,
+    user_comments: str = None,
 ) -> str:
     """Generate an article based on user-defined settings."""
     settings = {
@@ -104,10 +129,16 @@ def generate_article(
         "references": [r.strip() for r in references.split(",")],
         "num_examples": num_examples,
         "conclusion_type": conclusion_type,
+        "generated_article": generated_article,
+        "user_comments": user_comments,
     }
 
-    prompt = generate_prompt(settings)
+    if generated_article:
+        prompt = regenerate_prompt(settings)
+    else:
+        prompt = generate_prompt(settings)
 
+    print(prompt)
     if ai_model in ["OpenAI GPT 3.5", "OpenAI GPT 4"]:
         response = openai.ChatCompletion.create(
             model="gpt-4" if ai_model == "OpenAI GPT 4" else "gpt-3.5-turbo",
@@ -205,7 +236,7 @@ def format_references(text: str) -> str:
     in_references = False
 
     for line in lines:
-        if line.strip().lower() == "references":
+        if "references" in line.strip().lower():
             in_references = True
             continue
         if in_references:
@@ -215,6 +246,8 @@ def format_references(text: str) -> str:
 
     formatted_refs = []
     for i, ref in enumerate(references, 1):
+        if f"[{i}]" in ref:
+            ref = ref.replace(f"[{i}]", "")
         formatted_refs.append(f"[{i}] {ref}\n")
 
     return "\n\n".join(article_text) + "\n\nReferences:\n" + "\n".join(formatted_refs)
@@ -235,6 +268,8 @@ def generate_and_format(
     conclusion_type,
     ai_model,
     api_key,
+    generated_article: str = None,
+    user_comments: str = None,
 ):
     article = generate_article(
         topic,
@@ -251,6 +286,8 @@ def generate_and_format(
         conclusion_type,
         ai_model,
         api_key,
+        generated_article,
+        user_comments,
     )
     return format_references(article)
 
@@ -264,7 +301,9 @@ def create_interface():
         theme=gr.themes.Default(
             primary_hue=gr.themes.colors.pink, secondary_hue=gr.themes.colors.yellow, neutral_hue=gr.themes.colors.gray
         ),
-        css=".custom-textbox textarea { white-space: pre-wrap; }",
+        css="""
+            .input-highlight-pink block_label {background-color: #008080}
+            """,
     ) as demo:
         gr.Markdown("# Polygraf AI Content Writer", elem_classes="text-center text-3xl mb-6")
 
@@ -408,7 +447,10 @@ 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)
                 with gr.Row():
                     with gr.Column():
                         ai_detector_dropdown = gr.Radio(
@@ -419,9 +461,18 @@ def create_interface():
 
                 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", height="24em")
+                humanized_output = gr.Markdown(label="Humanized Article", value="\n\n\n\n", height="24em", render=True)
                 copy_to_input_btn = gr.Button("Copy to Input for AI Check")
 
+        def become_visible(text):
+            if text:
+                return gr.update(visible=True)
+            else:
+                return gr.update(visible=False)
+
+        output_article.change(become_visible, inputs=output_article, outputs=ai_comments)
+        ai_comments.change(become_visible, inputs=output_article, outputs=regenerate_btn)
+
         with gr.Accordion("Advanced Humanizer Settings", open=False):
             with gr.Row():
                 model_dropdown = gr.Radio(
@@ -467,6 +518,28 @@ def create_interface():
             ],
             outputs=[output_article],
         )
+        regenerate_btn.click(
+            fn=generate_and_format,
+            inputs=[
+                input_topic,
+                input_keywords,
+                input_length,
+                input_format,
+                input_writing_style,
+                input_tone,
+                input_user_category,
+                input_depth,
+                input_structure,
+                input_references,
+                input_num_examples,
+                input_conclusion,
+                ai_generator,
+                input_api,
+                output_article,
+                ai_comments,
+            ],
+            outputs=[output_article],
+        )
 
         ai_check_btn.click(
             fn=ai_check,