add warnings and errors

app.py

@@ -117,18 +117,38 @@ def save_model(model, model_name, quant_type_4, double_quant_4, compute_type_4,
 
 def quantize_and_save(profile: gr.OAuthProfile | None, oauth_token: gr.OAuthToken | None, model_name, quant_type_4, double_quant_4, compute_type_4, quant_storage_4, quantized_model_name, public):
     if oauth_token is None : 
-        return "Error : Please Sign In to your HuggingFace account to use the quantizer"
+        return """
+        <div class="error-box">
+            <h3>❌ Authentication Error</h3>
+            <p>Please sign in to your HuggingFace account to use the quantizer.</p>
+        </div>
+        """
     if not profile:
-        return "Error: Please Sign In to your HuggingFace account to use the quantizer"
+        return """
+        <div class="error-box">
+            <h3>❌ Authentication Error</h3>
+            <p>Please sign in to your HuggingFace account to use the quantizer.</p>
+        </div>
+        """
     exists_message = check_model_exists(oauth_token, profile.username, model_name, quantized_model_name)
     if exists_message : 
-        return exists_message
-    # try:
-    quantized_model = quantize_model(model_name, quant_type_4, double_quant_4, compute_type_4, quant_storage_4, oauth_token)
-    return save_model(quantized_model, model_name, quant_type_4, double_quant_4, compute_type_4, quant_storage_4, profile.username, oauth_token, quantized_model_name, public)
-    # except Exception as e : 
-    #     print(e)
-    #     return f"An error occurred: {str(e)}"
+        return f"""
+        <div class="warning-box">
+            <h3>⚠️ Model Already Exists</h3>
+            <p>{exists_message}</p>
+        </div>
+        """
+    try:
+        quantized_model = quantize_model(model_name, quant_type_4, double_quant_4, compute_type_4, quant_storage_4, oauth_token)
+        return save_model(quantized_model, model_name, quant_type_4, double_quant_4, compute_type_4, quant_storage_4, profile.username, oauth_token, quantized_model_name, public)
+    except Exception as e : 
+        print(e)
+        return f"""
+        <div class="error-box">
+            <h3>❌ Error Occurred</h3>
+            <p>{str(e)}</p>
+        </div>
+        """
 
 
 css="""/* Custom CSS to allow scrolling */