Update app.py

app.py

@@ -33,18 +33,17 @@ def process_message(message, history, analysis_prompt, rethinking_prompt, refine
 
     llama_prompt = f"{rethinking_prompt}\n\nConversation history:\n{context}\n\nOriginal user query: {message}\n\nInitial response: {' '.join(gpt4o_response)}\n\nPlease review and suggest improvements or confirm if satisfactory."
     llama_response = get_llm_response(llama_prompt, "gpt-4o-mini")
-    yield from stream_words("\nRethinking: ", llama_response)
+    yield from stream_words("\n\nRethinking: ", llama_response)
 
     if "<error>" in " ".join(llama_response):
-        return " ".join(gpt4o_response)
+        return
 
     if "done" not in " ".join(llama_response).lower():
         final_gpt4o_prompt = f"{refinement_prompt}\n\nConversation history:\n{context}\n\nOriginal user query: {message}\n\nInitial response: {' '.join(gpt4o_response)}\n\nSuggestion: {' '.join(llama_response)}\n\nPlease provide a final response considering the suggestion."
         final_response = get_llm_response(final_gpt4o_prompt, "gpt-4o-mini")
-        yield from stream_words("\nFinal Response: ", final_response)
-        return " ".join(final_response)
+        yield from stream_words("\n\nFinal Response: ", final_response)
     else:
-        return " ".join(gpt4o_response)
+        yield "\n\nFinal Response: The initial response is satisfactory and no further refinement is needed."
 
 def stream_words(prefix, words):
     response = prefix
@@ -54,14 +53,12 @@ def stream_words(prefix, words):
         yield response
 
 def respond(message, history, analysis_prompt, rethinking_prompt, refinement_prompt):
-    response = ""
+    full_response = ""
     for chunk in process_message(message, history, analysis_prompt, rethinking_prompt, refinement_prompt):
-        response = chunk
-        yield response
+        full_response = chunk
+        yield full_response
     
-    # Extract the final response from the last chunk
-    final_response = response.split("Final Response: ")[-1] if "Final Response: " in response else response.split("Analysis: ")[-1]
-    return final_response
+    return full_response
 
 # (The rest of the code remains the same: analysis_prompt, rethinking_prompt, refinement_prompt, and the Gradio
 # (Previous code remains the same)