Show iteration count and time used

app.py

@@ -67,6 +67,8 @@ def chill_out(text):
             <li>Faithfulness score: {result['faithfulness_score']:.0%}</li>
             <li>Spicy score: {result['spicy_score']:.0%}</li>
             <li>Overall score: {result['overall_score']:.0%}</li>
+            <li>Iterations: {result['iteration_count'] / result['max_allowed_iterations']:.0%} {result['iteration_count']} of max allowed iterations: {result['max_allowed_iterations']}</li> 
+            <li>Time used: {result['time_used']:.2f} seconds</li>
             <li>LLM requests made: {result['request_count']}</li>
         </ul>
     </div>

chill.py

@@ -42,7 +42,7 @@ global suggestions
 suggestions = []
 last_edit = ""
 request_count = 0
-start_time = time.time()
+start_time = None
 
 
 def improve_text_attempt():
@@ -125,6 +125,7 @@ def improvement_loop(input_text):
     global suggestions
     global request_count
     global start_time
+    iteration_count = 0
     suggestions = []
     last_edit = ""
     request_count = 0
@@ -133,6 +134,7 @@ def improvement_loop(input_text):
     original_text = input_text
 
     for iteration in range(1, max_iterations + 1):
+        iteration_count = iteration
         try:
             if iteration % 2 == 1:
                 last_edit = improve_text_attempt()
@@ -154,6 +156,9 @@ def improvement_loop(input_text):
             print("ValueError:", e)
             continue
     assert len(suggestions) > 0
+    suggestions[0]["iteration_count"] = iteration_count
+    suggestions[0]["max_allowed_iterations"] = max_iterations
+    suggestions[0]["time_used"] = time_used
     return suggestions[0]
 
 

gradio_cached_examples/14/log.csv

@@ -2,26 +2,32 @@ output,flag,username,timestamp
 "
     <div>
         <h4>Edited text:</h4>
-        <p>I am concerned that our current pace might delay the project's completion.</p>
+        <p>It seems we're moving a bit slower than anticipated. I'm concerned we might not meet our shipping deadline.</p>
         <h4>Details:</h4>
         <ul>
-            <li>Critique: The revised text does not convey the same level of urgency as the original.</li>
-            <li>Faithfulness Score: 0.85</li>
-            <li>Spicy Score: 0.2</li>
-            <li>Overall Score: 0.88</li>
+            <li>Critique: This is well-phrased and less inflammatory than the original. However, it might be more direct and still polite to say: 'We're behind schedule. I'm concerned we might not make our shipping deadline.'</li>
+            <li>Faithfulness score: 95%</li>
+            <li>Spicy score: 10%</li>
+            <li>Overall score: 97%</li>
+            <li>Iterations: 33% 2 of max allowed iterations: 6</li> 
+            <li>Time used: 2.24 seconds</li>
+            <li>LLM requests made: 4</li>
         </ul>
     </div>
-    ",,,2024-02-29 13:07:52.046527
+    ",,,2024-02-29 17:35:32.665693
 "
     <div>
         <h4>Edited text:</h4>
-        <p>I think your idea of a balanced diet may need some adjustments.</p>
+        <p>It seems your understanding of a balanced diet differs from mine, as it appears to include biscuits quite often.</p>
         <h4>Details:</h4>
         <ul>
-            <li>Critique: This critique is too vague and does not address the specific issues with the provided text.</li>
-            <li>Faithfulness Score: 0.5</li>
-            <li>Spicy Score: 0</li>
-            <li>Overall Score: 0.5</li>
+            <li>Critique: This critique is more subtle and less confrontational, which is good for preventing inflaming discussions. However, it could be more concise and closer to the original semantic intent. A possible revision could be: 'Your concept of a balanced diet appears to involve biscuits more often than usual.'</li>
+            <li>Faithfulness score: 95%</li>
+            <li>Spicy score: 10%</li>
+            <li>Overall score: 97%</li>
+            <li>Iterations: 33% 2 of max allowed iterations: 6</li> 
+            <li>Time used: 2.36 seconds</li>
+            <li>LLM requests made: 4</li>
         </ul>
     </div>
-    ",,,2024-02-29 13:08:14.666888
+    ",,,2024-02-29 17:35:35.029163