Update openai_client.py

openai_client.py

@@ -58,20 +58,17 @@ def refactor_code(code_snippet):
                 },
             ],
         )
-        refactor = response.text if hasattr(response, "text") else str(response)
-
         # Check if feedback is a Message object and extract text if necessary
         if hasattr(response, "content") and isinstance(response.content, list):
-            refactor = "\n\n".join(
-                [
-                    text_block.text
-                    for text_block in response.content
-                    if hasattr(text_block, "text")
-                ]
+            # Join the text blocks into a single string with line breaks
+            refactor = "\n".join(
+                [text_block.text for text_block in response.content if hasattr(text_block, "text")]
             )
+        else:
+            refactor = response.text if hasattr(response, "text") else str(response)
 
-        # return refactor
-        return st.code(refactor, language='python')
+        # Return the formatted string, ensuring it maintains line breaks
+        return refactor.strip()
     except Exception as e:
         return "Sorry, an error occurred while refactoring your code. Please try again later."