Update app.py

app.py

@@ -460,6 +460,33 @@ async def edit_suffix(chunk_id, suffix):
     result = api.editor.add_suffix(chunk_id, suffix)
     return result, api.editor.diff(chunk_id, result)
 
+async def generate_and_edit(source_text, target_start, target_end, response_prompt):
+    # Step 1: Generate source paragraph/code
+    memory = ConversationMemory()
+    chunk_id = memory.add_chunk(source_text, "user")
+    logger.info(f"Generated source chunk: {chunk_id}")
+
+    # Step 2: Cut out the target text
+    api = OpenAIApi(api_key=os.getenv("AZURE_OPENAI_API_KEY"))
+    cut_result = api.editor.cut(chunk_id, target_start, target_end)
+    logger.info(f"Cut target text from chunk: {chunk_id}, start: {target_start}, end: {target_end}")
+
+    # Step 3: Generate response
+    response = await api.fetch_response(response_prompt)
+    if "error" not in response:
+        response_text = response["content"]
+        logger.info(f"Generated response: {response_text}")
+    else:
+        return "Error: Failed to generate response", ""
+
+    # Step 4: Paste response into the target hole
+    paste_result = api.editor.paste(chunk_id, target_start)
+    logger.info(f"Pasted response into chunk: {chunk_id}, position: {target_start}")
+
+    # Return updated text and diff
+    diff = api.editor.diff(chunk_id, paste_result) if "Error" not in paste_result else ""
+    return paste_result, diff
+
 def create_ui():
     with gr.Blocks(title="Azure OpenAI Chat & Text Editor") as demo:
         gr.Markdown("# Azure OpenAI Chat with Text Editing")
@@ -513,6 +540,20 @@ def create_ui():
             prefix_btn.click(fn=edit_prefix, inputs=[chunk_id, prefix], outputs=[chunk_text, diff_output])
             suffix_btn.click(fn=edit_suffix, inputs=[chunk_id, suffix], outputs=[chunk_text, diff_output])
         
+        with gr.Tab("Advanced Text Manipulation"):
+            source_text = gr.Textbox(label="Source Text", value="This is a sample paragraph. [TARGET] This is the rest of the text.")
+            target_start = gr.Number(label="Target Start Index", value=21, precision=0)
+            target_end = gr.Number(label="Target End Index", value=28, precision=0)
+            response_prompt = gr.Textbox(label="Response Prompt", value="Generate a response for the target section.")
+            generate_btn = gr.Button("Generate and Edit")
+            result_text = gr.Textbox(label="Result Text", interactive=False)
+            result_diff = gr.Textbox(label="Result Diff", interactive=False)
+            generate_btn.click(
+                fn=generate_and_edit,
+                inputs=[source_text, target_start, target_end, response_prompt],
+                outputs=[result_text, result_diff]
+            )
+        
         with gr.Tab("Logs"):
             logs = gr.Textbox(label="Application Logs", interactive=False)
             logs_btn = gr.Button("Refresh Logs")