Update styles.py

styles.py

@@ -82,28 +82,6 @@ def get_custom_css():
             border-radius: 5px;
         }
         
-        /* Fixed styling for model response area */
-        .element-container:has(h1:contains("Oferta Generada")) ~ div {
-            border: 2px solid #3D89B8;
-            border-radius: 8px;
-            padding: 15px;
-            margin: 10px 0;
-            box-shadow: 0 2px 5px rgba(61, 137, 184, 0.1);
-        }
-        
-        /* Alternative selector for response area */
-        .row-widget.stDownloadButton {
-            margin-top: 15px;
-            margin-bottom: 15px;
-        }
-        
-        /* Make sure the heading stands out */
-        h3:contains("Oferta Generada") {
-            padding-bottom: 10px;
-            border-bottom: 1px solid #3D89B8;
-            margin-bottom: 15px;
-        }
-        
         /* Style for file uploader */
         .stFileUploader > div {
             border: 1px dashed #3D89B8;
@@ -127,5 +105,25 @@ def get_custom_css():
             color: #1A3A5F;
             font-weight: bold;
         }
+        
+        /* Custom styling for response container */
+        .response-container {
+            padding: 15px;
+            border: 2px solid #3D89B8;
+            border-radius: 8px;
+            margin: 10px 0;
+            box-shadow: 0 2px 5px rgba(61, 137, 184, 0.1);
+        }
     </style>
+    """
+
+def get_response_html_wrapper(content):
+    """
+    Wraps content in a styled div with COPYXPERT branding
+    """
+    return f"""
+    <div style="padding: 15px; border: 2px solid #3D89B8; border-radius: 8px; box-shadow: 0 2px 5px rgba(61, 137, 184, 0.1);">
+        <h3 style="color: #1A3A5F; padding-bottom: 10px; border-bottom: 1px solid #3D89B8; margin-bottom: 15px;">Oferta Generada</h3>
+        <p>{content}</p>
+    </div>
     """